Save the date! Google I/O returns May 18-20 Register now

Module: oryx.core.pytree

Contains the Pytree class.

Classes

class Pytree: Class that registers objects as Jax pytree_nodes.