Google I/O is a wrap! Catch up on TensorFlow sessions View sessions

oryx.core.trace_util.pv_like

Converts a JAX value type into a JAX PartialVal.