OpTree: Optimized PyTree Utilities
-
Updated
Jul 7, 2026 - Python
#
pytree
Here are 15 public repositories matching this topic...
Visualize, create, and operate on pytrees in the most intuitive way possible.
-
Updated
Jan 11, 2025 - Python
A stateful pytree library for training neural networks.
-
Updated
Aug 31, 2025 - Python
Save pytrees efficiently in hdf5 files.
-
Updated
Jul 6, 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
-
Updated
Apr 10, 2025 - 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
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
Improve this page
Add a description, image, and links to the pytree topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the pytree topic, visit your repo's landing page and select "manage topics."