ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

oryx.core.Pytree

Class that registers objects as Jax pytree_nodes.

Methods

flatten

View source

unflatten

View source