oryx.core.Pytree

Class that registers objects as Jax pytree_nodes.

Methods

flatten

View source

unflatten

View source