oryx.core.tie_in

A reimplementation of jax.tie_in that handles pytrees.