pytree topic
List
pytree repositories
treeo
59
Stars
4
Forks
Watchers
A small library for creating and manipulating custom JAX Pytree classes
pytreeclass
38
Stars
1
Forks
Watchers
Visualize, create, and operate on pytrees in the most intuitive way possible.
pax
21
Stars
1
Forks
Watchers
A stateful pytree library for training neural networks.
optree
110
Stars
5
Forks
Watchers
OpTree: Optimized PyTree Utilities