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

Module: oryx.core.pytree

Contains the Pytree class.

Classes

class Pytree: Class that registers objects as Jax pytree_nodes.