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.
Proceed to this step ONL Y after all prior tests are passing.Stop and wait for user approval
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.