Searching protocol for "pytree"
Effortless JAX/Equinox config setup.
JAX+Equinox numerics, standardized.
Accelerate ML with JAX: fast, differentiable.
High-performance NumPy-like computing with JAX.