12 projects
ntx
JAX-native monoenergetic neoclassical transport solver
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.
magpylib-jax
Differentiable JAX-native magnetic field modeling with Magpylib-compatible APIs
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
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
