ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

oryx.core.trace_util.pv_like

Converts a JAX value type into a JAX PartialVal.