vmec_jax.mercier

Differentiable Mercier and JXBFORCE algebra kernels.

These helpers are the source-level JAX counterparts of VMEC’s Mercier and JXBFORCE profile reductions. They intentionally avoid VMEC-state orchestration so they can be tested against the algebra directly and reused by finite-beta objectives.

Functions

jxbforce_profiles_from_realspace(*, phips, ...)

Return JAX-differentiable JXBFORCE 1D field/current profiles.

mercier_surface_integrals_from_realspace(*, ...)

Return JAX-differentiable Mercier surface integrals.

mercier_terms_from_profile_integrals(*, s, ...)

Return JAX-differentiable Mercier terms from 1D VMEC profile integrals.