OpTree: Optimized PyTree Utilities
-
Updated
Aug 29, 2026 - Python
#
pytree
Here are 15 public repositories matching this topic...
Visualize, create, and operate on pytrees in the most intuitive way possible.
-
Updated
Aug 18, 2026 - Python
A stateful pytree library for training neural networks.
-
Updated
Aug 31, 2025 - Python
Save pytrees efficiently in hdf5 files.
-
Updated
Jul 18, 2026 - Python
High performance queues for pytree objects.
-
Updated
Jan 2, 2023 - Python
"Module pytrees" that cleanly handle parameter trainability and transformations for JAX models.
-
Updated
May 11, 2023 - Python
practiced actionlib and py-trees ros tutorials
-
Updated
Dec 22, 2021 - Python
High-performance, branchless numerical stability kernels and compiler-optimized core infrastructure for advanced JAX/XLA deep learning architectures.
infrastructure machine-learning deep-learning computational-mathematics high-performance-computing gradient-descent backpropagation compiler-optimization numerical-stability geometric-deep-learning jax gradient-clipping xla branchless pytree optax gradient-vanishing concretization-type-error deepmind-optax jax-optimization
-
Updated
Jul 8, 2026 - Python
-
Updated
Apr 10, 2025 - Python
A functional and lightweight neural network library for JAX.
python machine-learning deep-learning functional-programming gpu neural-networks flax equinox tpu jax pytree
-
Updated
Jun 3, 2026 - Python
Add this topic to your repo
To associate your repository with the pytree topic, visit your repo's landing page and select "manage topics."