Here are
11 public repositories
matching this topic...
OpTree: Optimized PyTree Utilities
Updated
Mar 12, 2026
Python
A small library for creating and manipulating custom JAX Pytree classes
Updated
Feb 26, 2023
Python
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
High performance queues for pytree objects.
Updated
Jan 2, 2023
Python
practiced actionlib and py-trees ros tutorials
Updated
Dec 22, 2021
Python
"Module pytrees" that cleanly handle parameter trainability and transformations for JAX models.
Updated
May 11, 2023
Python
Updated
Apr 10, 2025
Python
io+hdf5+lazy-loading+processing-pipelines for EHRs and generic PyTrees for JAX friendly tasks, mainly ML
Updated
Oct 28, 2025
Python
Jax batching utility library
Updated
May 29, 2025
Python
Designing AI-Powered Codebase Intelligence Systems to Accelerate Developer Onboarding in Production Environments
Updated
Mar 12, 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.
Curate this topic
Add this topic to your repo
To associate your repository with the
pytree
topic, visit your repo's landing page and select "manage topics."
Learn more
You can’t perform that action at this time.