Module: oryx.core.pytree

Contains the Pytree class.

Classes

class Pytree: Class that registers objects as Jax pytree_nodes.