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

oryx.core.tie_in

A reimplementation of jax.tie_in that handles pytrees.