pytree

There are 0 repository under pytree topic.

metaopt / optree
OpTree: Optimized PyTree Utilities
pytree tree tree-structure tree-flatten optree tree-map tree-operation
Language:Python 120
cgarciae / treeo
A small library for creating and manipulating custom JAX Pytree classes
jax pytree
Language:Python 59
ASEM000 / pytreeclass
Visualize, create, and operate on pytrees in the most intuitive way possible.
dataclasses deep-learning jax machine-learning pytorch tensorflow pytree data pipelines
Language:Python 38
pax
NTT123 / pax
A stateful pytree library for training neural networks.
deep-learning jax opax pytree
Language:Python 21
thomashirtz / treequeues
High performance queues for pytree objects.
jax multiprocessing python pytree queues
Language:Python 4
JaxGaussianProcesses / mytree
"Module pytrees" that cleanly handle parameter trainability and transformations for JAX models.
jax pytree bijector trainability module
Language:Python 3
tejalbarnwal / behaviourtree_tutorial
practiced actionlib and py-trees ros tutorials
actionlib ros-action pytree ros
Language:Python 1
mansooralodhi / intervalAdjointSigAnalysis
msThesis
adjoint-method adjoint-sensitivities algorithmic-differentiation interval-arithmetic jax pytree
Language:Python 0
angelo-angonezi / pytree
A simpler 'tree' linux command for terminal file/folder visualization, running on Python
linux tree pytree terminal
Language:Python

metaopt / optree