A new JAX-based gyrokinetic code achieves statistical parity with the legacy GKW solver, delivers substantial speedup on GPUs, and enables ML workflows via automatic differentiation.
Y ou MUST fully load the following specific files into your context, and explore for any other useful ones: •gkw ref/src/gkw.f90(Main loop over large time steps, normalisation)
1 Pith paper cite this work. Polarity classification is still indexing.
1
Pith paper citing it
fields
physics.plasm-ph 1years
2026 1verdicts
UNVERDICTED 1representative citing papers
citing papers explorer
-
gyaradax: Local Gyrokinetics JAX Code
A new JAX-based gyrokinetic code achieves statistical parity with the legacy GKW solver, delivers substantial speedup on GPUs, and enables ML workflows via automatic differentiation.