13 projects
sfincs-jax
Neoclassical transport solver with CPU/GPU and differentiable JAX workflows
spectraxgk
JAX gyrokinetic solver with Hermite-Laguerre velocity space
vmec-jax
End-to-end differentiable JAX implementation of VMEC2000 for fixed and free-boundary equilibria.
ntx
JAX-native monoenergetic neoclassical transport solver
neo-jax
Neoclassical transport code for effective ripple and trapped-particle diagnostics in Boozer coordinates
booz-xform-jax
Pure Python JAX implementation of the Boozer coordinate transformation
virtual-casing-jax
JAX implementation of the virtual casing principle with high-order quadrature
magpylib-jax
Differentiable JAX-native magnetic field modeling with Magpylib-compatible APIs
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