Attend the Women in ML Symposium on December 7 Register now

oryx.core.Pytree

Stay organized with collections Save and categorize content based on your preferences.

Class that registers objects as Jax pytree_nodes.

Methods

flatten

View source

unflatten

View source