tree-math

github.comtool

Convert functions that operate on arrays into functions that operate on PyTrees.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   Equinox github.com

   tool category #jax link ↗

   Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX.

   
2. ▲
   TF2JAX github.com

   tool category #jax link ↗

   Convert functions/graphs to JAX functions.

   
3. ▲
   JAX github.com

   tool category #python-data-science link ↗

   Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more.

   
4. ▲
   Mctx github.com

   tool category #jax link ↗

   Monte Carlo tree search algorithms in native JAX.

   
5. ▲
   rtree github.com

   tool category #jvm link ↗

   Immutable in-memory R-tree and R*-tree implementations in Java with reactive api.

   
6. ▲
   Interaction Trees github.com

   tool category #coq link ↗

   Library for representing recursive and impure programs.

   

JAX › Libraries > New Libraries: “Convert functions that operate on arrays into functions that operate on PyTrees.”