vmec_jax.grids

Angle grids consistent with VMEC conventions.

VMEC typically computes on one field period. Internally the toroidal coordinate often behaves like zeta = NFP * phi_phys, such that zeta in [0, 2pi) spans one field period.

We keep grids explicit and leave symmetry reductions (ntheta2 / ntheta3) for later.

Functions

angle_steps(*, ntheta, nzeta)	Return uniform angle spacings (dtheta, dzeta) for periodic grids.
make_angle_grid(ntheta, nzeta, nfp[, endpoint])	Create theta, zeta grids.

Classes

AngleGrid(theta, zeta, nfp)