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