Google I/O returns May 18-20! Reserve space and build your schedule Register now

oryx.core.tie_in

A reimplementation of jax.tie_in that handles pytrees.