Literature and code context

MHX is positioned between validated plasma/MHD solvers and differentiable JAX research workflows.

Differentiable JAX ecosystem

  • JAX documentation motivates pure array programs that compose with jit, vmap, grad, and accelerator execution.

  • Diffrax adjoints document checkpointed discrete adjoints and continuous/backsolve adjoints for differentiable time integration.

  • Lineax provides JAX-native matrix-free linear solves, useful for implicit diffusion, projection, and elliptic pieces.

  • JAX-Fluids is a current reference for differentiable CFD implementation and performance discipline in JAX.

Plasma and MHD validation targets

The first validation sequence will cover FKR/Coppi tearing growth, plasmoid instability, ideal tearing, GEM-style Hall reconnection, and generalized Ohm’s law terms. Extended-MHD examples will be added only with explicit assumptions, equations, tests, and limitations.

For the active tearing validation gates:

For generalized Ohm’s law and collisionless/two-fluid reconnection context:

External comparison codes

MHX will document comparison workflows against public or widely used codes: