Last released Jun 2, 2026
Tree traversals using JAX
Last released Mar 5, 2024
Differential geometry using jax
Supported by