vmec_jax.static

Static (compile-time) data for vmec_jax.

This module defines a small “static” container that groups together data that should be precomputed once per equilibrium problem and then reused inside jax.jit’d kernels:

  • mode table (m,n)

  • angle grids (theta,zeta)

  • helical basis tensors (cos/sin on the grid)

  • radial grid (s in [0,1])

Keeping these pieces together prevents accidental recomputation and makes it easier to write fast, end-to-end differentiable kernels later.

Functions

build_static(cfg, *[, grid, mgrid_metadata, ...])

Build the VMECStatic container from a parsed config.

initial_free_boundary_state(cfg)

Lazily construct free-boundary runtime state.

Classes

VMECStatic(cfg, modes, grid, basis, s[, ...])

Precomputed static data for a VMEC run.