pytree
OpTree: Optimized PyTree Utilities
Visualize, create, and operate on pytrees in the most intuitive way possible.
HDF5 serialization for JAX and NumPy pytrees with full type fidelity and reference identity
A small library for creating and manipulating custom JAX Pytree classes
A functional and lightweight neural network library for JAX.
A stateful pytree library for training neural networks.
EHRs io+processing for JAX+equinox compatible tasks, e.g. mainly ML.
