Last released Aug 11, 2026
Differentiable numerical methods in JAX (FEM, FDM, spectral) with scientific machine learning
Last released Jun 1, 2026
Jax Neural Operators
Last released May 30, 2026
One-stop JAX foundation model repository
Supported by