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

oryx.core.Pytree

Class that registers objects as Jax pytree_nodes.

Methods

flatten

View source

unflatten

View source