oryx.core.tie_in

A reimplementation of jax.tie_in that handles pytrees.

Main aliases

oryx.core.primitive.tie_in