GKX

JAX gyrokinetics for tokamak and stellarator flux tubes, with a Hermite–Laguerre velocity basis, certified eigenvalues, nonlinear turbulence and exact derivatives. Start with Quickstart and Tutorial; Resolution, Step and Solver Guidance says what to set. Features and Claim Scope lists the features and what is and is not claimed; Design and Code Map explains why the code is built this way; Validation holds the evidence.

Getting started

• Quickstart
• Tutorial
• Resolution, Step and Solver Guidance

Inputs and outputs

• Input Files and Executable
• Saved Outputs and Restart Files

Examples

• Examples
• Stellarator optimization

Physics and models

• Theory
• Normalization
• Conventions and conversion to GX, GS2 and stella
• Geometry
• Linear Model And Derivations
• Operators And Terms
• Quasilinear Transport

Numerics and algorithms

• Numerics
• Solvers
• Methods and Decisions
• Differentiable matrix-free eigenmodes
• Nonlinear automatic differentiation
• Parallelization policy

How the code works

• Design and Code Map
• Numerical defaults from SOLVAX

Benchmarks and validation

• Validation
• Performance
• Related Codes
• Features and Claim Scope

Development

• Testing
• API Reference
• References