Save the date! Google I/O returns May 18-20 Register now

oryx.core.trace_util.pv_like

Converts a JAX value type into a JAX PartialVal.