vmec_jax.diagnostics

Lightweight diagnostic helpers.

These utilities are intentionally dependency-free (NumPy-only) and meant to print useful debugging information that you can copy/paste into chat.

We keep this module small and stable so it can be used from examples and tests without pulling in plotting libraries.

Functions

print_jacobian_stats(sqrtg, *[, indent])	Print useful statistics for the Jacobian sqrt(g).
print_summary(s, *[, indent])	Pretty-print a Summary.
slice_excluding_axis(a[, axis_dim])	Return a[1:] along the chosen axis (used to avoid s=0 degeneracy).
summarize_array(name, x, *[, q])	Return basic stats + quantiles for an array-like.
summarize_many(names_and_arrays, *[, indent])	Summarize many arrays.
vmec_internal_mn_from_state(state, static, *)	Return VMEC (m,n>=0) coefficient blocks from a signed-coefficient state.
vmec_xc_from_mn_blocks(*, rcc, rss, zsc, ...)	Pack VMEC (m,n>=0) coefficient blocks into the 1D xc vector.

Classes

Summary(name, shape, dtype, min, max, mean, ...)