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

oryx.core.Pytree

Class that registers objects as Jax pytree_nodes.

Methods

flatten

View source

unflatten

View source