Last released Sep 30, 2026
Advanced automatic-differentiation and batching utilities for JAX
Last released Jul 31, 2026
Computes, analyzes and optimizes 3D MHD equilibria for stellarators and tokamaks
Last released Jun 22, 2026
Fourier interpolation and function approximation with JAX
Last released May 21, 2026
Multistage neural networks with JAX