18 projects
booz-xform-jax
Pure Python JAX implementation of the Boozer coordinate transformation
vmex
JAX implementation of VMEC2000 with differentiable fixed-boundary and branch-local free-boundary research paths.
gkx
JAX gyrokinetic solver with Hermite-Laguerre velocity space
dkx
Neoclassical transport solver with CPU/GPU and differentiable JAX workflows
solvax
Differentiable structured linear solvers, preconditioners and matrix-free methods in JAX
virtual-casing-jax
JAX implementation of the virtual casing principle with high-order quadrature
ntx
JAX-native monoenergetic neoclassical transport solver
magpylib-jax
Differentiable JAX-native magnetic field modeling with Magpylib-compatible APIs
drbx
JAX-first edge and scrape-off-layer plasma code with differentiable, hardware-agnostic solver infrastructure.
spectraxgk
JAX gyrokinetic solver with Hermite-Laguerre velocity space
sfincs-jax
Neoclassical transport solver with CPU/GPU and differentiable JAX workflows
vmec-jax
JAX implementation of VMEC2000 with differentiable fixed-boundary and branch-local free-boundary research paths.
neo-jax
Neoclassical transport code for effective ripple and trapped-particle diagnostics in Boozer coordinates
spectrax
SPECTRAX: Hermite-Fourier Vlasov solver in JAX
qicna
Quasi-isodynamic Stellarator Construction
jaxincell
JAX-in-Cell: Particle-in-Cell code in JAX
flexipde
Flexible high‑performance plasma physics solver in Python using JAX and NumPy
essos
ESSOS: e-Stellarator Simulation and Optimization Suite
