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

oryx.core.tie_in

A reimplementation of jax.tie_in that handles pytrees.