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