6 projects
jaxincell
JAX-in-Cell: Particle-in-Cell code in JAX
flexipde
Flexible high‑performance plasma physics solver in Python using JAX and NumPy
spectraxgk
Fourier–Hermite and DG solvers for multispecies Gyrokinetic equation with JAX.
essos
ESSOS: e-Stellarator Simulation and Optimization Suite
qicna
Quasi-isodynamic Stellarator Construction
spectrax
SPECTRAX: Hermite-Fourier Vlasov solver in JAX