LICENSE
MANIFEST.in
README.md
pyproject.toml
benchmarks/README.md
benchmarks/basis_orthonormality.py
benchmarks/cyclone_linear_benchmark.py
benchmarks/etg_linear_benchmark.py
benchmarks/kbm_beta_scan.py
benchmarks/kinetic_linear_benchmark.py
benchmarks/runtime_miller_zonal_response.toml
benchmarks/runtime_secondary_slab.toml
benchmarks/runtime_w7x_zonal_response_vmec.toml
benchmarks/secondary_slab_workflow.py
benchmarks/tem_linear_benchmark.py
benchmarks/results/README.md
benchmarks/results/manifest.toml
docs/algorithms.rst
docs/api.rst
docs/architecture.rst
docs/architecture_refactor_plan.rst
docs/benchmarks.rst
docs/code_structure.rst
docs/codes.rst
docs/conf.py
docs/differentiable_refactor_plan.rst
docs/examples.rst
docs/geometry.rst
docs/index.rst
docs/inputs.rst
docs/linear_model.rst
docs/manuscript_figures.rst
docs/normalization.rst
docs/numerics.rst
docs/operators.rst
docs/outputs.rst
docs/parallelization.rst
docs/performance.rst
docs/quasilinear.rst
docs/quickstart.rst
docs/references.rst
docs/release_scope.rst
docs/roadmap.rst
docs/solvers.rst
docs/stellarator_optimization.rst
docs/testing.rst
docs/theory.rst
docs/validation_strategy.rst
docs/verification_matrix.rst
examples/linear/axisymmetric/cyclone.toml
examples/linear/axisymmetric/cyclone_runtime_linear.py
examples/linear/axisymmetric/etg.toml
examples/linear/axisymmetric/etg_linear_auto.py
examples/linear/axisymmetric/etg_runtime_linear.py
examples/linear/axisymmetric/kaw_runtime_linear.py
examples/linear/axisymmetric/kbm_runtime_linear.py
examples/linear/axisymmetric/runtime_circular_vmec_linear.toml
examples/linear/axisymmetric/runtime_cyclone.toml
examples/linear/axisymmetric/runtime_cyclone_miller_quasilinear.toml
examples/linear/axisymmetric/runtime_cyclone_quasilinear.toml
examples/linear/axisymmetric/runtime_etg.toml
examples/linear/axisymmetric/runtime_kaw.toml
examples/linear/axisymmetric/runtime_kbm.toml
examples/linear/non-axisymmetric/hsx_linear_imported_geometry.py
examples/linear/non-axisymmetric/runtime_hsx_linear_quasilinear.toml
examples/linear/non-axisymmetric/runtime_w7x_linear_imported_geometry.toml
examples/linear/non-axisymmetric/runtime_w7x_linear_quasilinear_vmec.toml
examples/linear/non-axisymmetric/w7x_linear_imported_geometry.py
examples/nonlinear/axisymmetric/cetg_runtime_nonlinear.py
examples/nonlinear/axisymmetric/cyclone_runtime_nonlinear.py
examples/nonlinear/axisymmetric/etg_runtime_nonlinear.py
examples/nonlinear/axisymmetric/kbm_runtime_nonlinear.py
examples/nonlinear/axisymmetric/miller_nonlinear_runtime.py
examples/nonlinear/axisymmetric/runtime_cetg_reference.toml
examples/nonlinear/axisymmetric/runtime_circular_vmec_nonlinear.toml
examples/nonlinear/axisymmetric/runtime_cyclone_nonlinear.toml
examples/nonlinear/axisymmetric/runtime_cyclone_nonlinear_miller.toml
examples/nonlinear/axisymmetric/runtime_cyclone_nonlinear_short.toml
examples/nonlinear/axisymmetric/runtime_cyclone_nonlinear_t400.toml
examples/nonlinear/axisymmetric/runtime_etg_nonlinear.toml
examples/nonlinear/axisymmetric/runtime_kbm_nonlinear.toml
examples/nonlinear/axisymmetric/runtime_kbm_nonlinear_seed.toml
examples/nonlinear/axisymmetric/runtime_kbm_nonlinear_short.toml
examples/nonlinear/axisymmetric/runtime_kbm_nonlinear_short_lockin.toml
examples/nonlinear/axisymmetric/runtime_kbm_nonlinear_t100.toml
examples/nonlinear/axisymmetric/runtime_kbm_nonlinear_t100_nx4ny8_dt9e4.toml
examples/nonlinear/non-axisymmetric/hsx_nonlinear_imported_geometry.py
examples/nonlinear/non-axisymmetric/hsx_nonlinear_vmec_geometry.py
examples/nonlinear/non-axisymmetric/reference_hsx_nonlinear_adiabatic_electrons.toml
examples/nonlinear/non-axisymmetric/runtime_hsx_nonlinear_vmec_geometry.toml
examples/nonlinear/non-axisymmetric/runtime_w7x_nonlinear_imported_geometry.toml
examples/nonlinear/non-axisymmetric/runtime_w7x_nonlinear_vmec_geometry.toml
examples/nonlinear/non-axisymmetric/w7x_nonlinear_imported_geometry.py
examples/nonlinear/non-axisymmetric/w7x_nonlinear_vmec_geometry.py
examples/optimization/QA_optimization_linear_ITG.py
examples/optimization/QA_optimization_nonlinear_ITG.py
examples/optimization/QA_optimization_quasilinear_ITG.py
examples/optimization/QA_parameter_scan.py
examples/parallelization/independent_ky_runtime_batch_scan.py
examples/parallelization/runtime_batch_ky_scan.toml
examples/theory_and_demos/autodiff_inverse_growth.py
examples/theory_and_demos/autodiff_inverse_twomode.py
examples/theory_and_demos/cyclone_geometry.py
examples/theory_and_demos/differentiable_geometry_bridge.py
examples/theory_and_demos/diffrax_linear_demo.py
examples/theory_and_demos/example.py
examples/theory_and_demos/gradB_coupling_hl_1d.py
examples/theory_and_demos/linear_rhs_demo.py
examples/theory_and_demos/quasilinear_implicit_sensitivity.py
examples/theory_and_demos/two_stream_hermite_1d.py
examples/theory_and_demos/reduced_stellarator_itg/_stellarator_itg_plotting.py
examples/theory_and_demos/reduced_stellarator_itg/_stellarator_itg_workflow.py
examples/theory_and_demos/reduced_stellarator_itg/compare_stellarator_itg_optimizations.py
examples/theory_and_demos/reduced_stellarator_itg/stellarator_itg_growth_optimization.py
examples/theory_and_demos/reduced_stellarator_itg/stellarator_itg_nonlinear_heat_flux_optimization.py
examples/theory_and_demos/reduced_stellarator_itg/stellarator_itg_portfolio_gate.py
examples/theory_and_demos/reduced_stellarator_itg/stellarator_itg_quasilinear_flux_optimization.py
examples/utilities/plot_runtime_outputs.py
examples/utilities/runtime_from_toml.py
examples/utilities/strong_scaling_sweep.py
src/spectraxgk/__init__.py
src/spectraxgk/_version.py
src/spectraxgk/benchmarks.py
src/spectraxgk/cli.py
src/spectraxgk/config.py
src/spectraxgk/linear.py
src/spectraxgk/nonlinear.py
src/spectraxgk/quasilinear.py
src/spectraxgk/runtime.py
src/spectraxgk.egg-info/PKG-INFO
src/spectraxgk.egg-info/SOURCES.txt
src/spectraxgk.egg-info/dependency_links.txt
src/spectraxgk.egg-info/entry_points.txt
src/spectraxgk.egg-info/requires.txt
src/spectraxgk.egg-info/top_level.txt
src/spectraxgk/api/__init__.py
src/spectraxgk/api/artifacts.py
src/spectraxgk/api/benchmarks.py
src/spectraxgk/api/configuration.py
src/spectraxgk/api/diagnostics.py
src/spectraxgk/api/geometry.py
src/spectraxgk/api/objectives.py
src/spectraxgk/api/parallel.py
src/spectraxgk/api/runtime.py
src/spectraxgk/api/solvers.py
src/spectraxgk/api/validation.py
src/spectraxgk/artifacts/__init__.py
src/spectraxgk/artifacts/benchmark_plots.py
src/spectraxgk/artifacts/diagnostic_plots.py
src/spectraxgk/artifacts/io.py
src/spectraxgk/artifacts/linear.py
src/spectraxgk/artifacts/nonlinear.py
src/spectraxgk/artifacts/nonlinear_diagnostics.py
src/spectraxgk/artifacts/nonlinear_netcdf.py
src/spectraxgk/artifacts/nonlinear_netcdf_diagnostics.py
src/spectraxgk/artifacts/nonlinear_netcdf_fields.py
src/spectraxgk/artifacts/nonlinear_netcdf_geometry.py
src/spectraxgk/artifacts/plot_style.py
src/spectraxgk/artifacts/plotting.py
src/spectraxgk/artifacts/restart.py
src/spectraxgk/artifacts/runtime_plots.py
src/spectraxgk/artifacts/spectral_layout.py
src/spectraxgk/artifacts/validation.py
src/spectraxgk/artifacts/zonal_plots.py
src/spectraxgk/core/__init__.py
src/spectraxgk/core/contracts.py
src/spectraxgk/core/extension_points.py
src/spectraxgk/core/grid.py
src/spectraxgk/core/species.py
src/spectraxgk/core/velocity.py
src/spectraxgk/data/cyclone_reference_adiabatic.csv
src/spectraxgk/data/cyclone_reference_kinetic.csv
src/spectraxgk/data/etg_reference.csv
src/spectraxgk/data/kbm_reference.csv
src/spectraxgk/data/tem_reference.csv
src/spectraxgk/diagnostics/__init__.py
src/spectraxgk/diagnostics/analysis.py
src/spectraxgk/diagnostics/channels.py
src/spectraxgk/diagnostics/core.py
src/spectraxgk/diagnostics/energy.py
src/spectraxgk/diagnostics/growth_fit.py
src/spectraxgk/diagnostics/growth_rates.py
src/spectraxgk/diagnostics/growth_series.py
src/spectraxgk/diagnostics/growth_windows.py
src/spectraxgk/diagnostics/metadata.py
src/spectraxgk/diagnostics/modes.py
src/spectraxgk/diagnostics/normalization.py
src/spectraxgk/diagnostics/quasilinear_transport.py
src/spectraxgk/diagnostics/resolved.py
src/spectraxgk/diagnostics/transport.py
src/spectraxgk/diagnostics/weights.py
src/spectraxgk/geometry/__init__.py
src/spectraxgk/geometry/analytic.py
src/spectraxgk/geometry/autodiff_checks.py
src/spectraxgk/geometry/backend_discovery.py
src/spectraxgk/geometry/booz_xform_bridge.py
src/spectraxgk/geometry/boundaries.py
src/spectraxgk/geometry/core.py
src/spectraxgk/geometry/differentiable.py
src/spectraxgk/geometry/flux_tube.py
src/spectraxgk/geometry/flux_tube_contract.py
src/spectraxgk/geometry/miller.py
src/spectraxgk/geometry/miller_eik.py
src/spectraxgk/geometry/numerics.py
src/spectraxgk/geometry/sensitivity.py
src/spectraxgk/geometry/vmec_boozer_constants.py
src/spectraxgk/geometry/vmec_boozer_core.py
src/spectraxgk/geometry/vmec_boozer_derivatives.py
src/spectraxgk/geometry/vmec_boundary_chain.py
src/spectraxgk/geometry/vmec_boundary_collection.py
src/spectraxgk/geometry/vmec_eik.py
src/spectraxgk/geometry/vmec_field_line_sampling.py
src/spectraxgk/geometry/vmec_flux_tube_reports.py
src/spectraxgk/geometry/vmec_state_controls.py
src/spectraxgk/geometry/vmec_state_sensitivity.py
src/spectraxgk/geometry/vmec_tensor_mapping.py
src/spectraxgk/geometry_backends/__init__.py
src/spectraxgk/geometry_backends/kernels.py
src/spectraxgk/geometry_backends/miller.py
src/spectraxgk/geometry_backends/miller_core.py
src/spectraxgk/geometry_backends/miller_io.py
src/spectraxgk/geometry_backends/miller_numerics.py
src/spectraxgk/geometry_backends/miller_pipeline.py
src/spectraxgk/geometry_backends/miller_profiles.py
src/spectraxgk/geometry_backends/vmec.py
src/spectraxgk/geometry_backends/vmec_backend_discovery.py
src/spectraxgk/geometry_backends/vmec_fieldline_numerics.py
src/spectraxgk/geometry_backends/vmec_fieldlines.py
src/spectraxgk/geometry_backends/vmec_io.py
src/spectraxgk/geometry_backends/vmec_numerics.py
src/spectraxgk/geometry_backends/vmec_pipeline.py
src/spectraxgk/geometry_backends/vmec_remap.py
src/spectraxgk/geometry_backends/vmec_splines.py
src/spectraxgk/geometry_backends/vmec_types.py
src/spectraxgk/objectives/__init__.py
src/spectraxgk/objectives/core.py
src/spectraxgk/objectives/eigen.py
src/spectraxgk/objectives/geometry.py
src/spectraxgk/objectives/gradient_gates.py
src/spectraxgk/objectives/nonlinear_window.py
src/spectraxgk/objectives/portfolio_artifacts.py
src/spectraxgk/objectives/portfolio_contracts.py
src/spectraxgk/objectives/portfolio_sensitivity.py
src/spectraxgk/objectives/qa_low_turbulence.py
src/spectraxgk/objectives/qa_low_turbulence_artifacts.py
src/spectraxgk/objectives/qa_low_turbulence_contracts.py
src/spectraxgk/objectives/qa_low_turbulence_model.py
src/spectraxgk/objectives/qa_low_turbulence_optimizer.py
src/spectraxgk/objectives/qa_low_turbulence_residuals.py
src/spectraxgk/objectives/sampling.py
src/spectraxgk/objectives/solver_gradient_reports.py
src/spectraxgk/objectives/solver_gradients.py
src/spectraxgk/objectives/solver_vmec.py
src/spectraxgk/objectives/stellarator.py
src/spectraxgk/objectives/stellarator_contracts.py
src/spectraxgk/objectives/stellarator_reduced.py
src/spectraxgk/objectives/stellarator_residuals.py
src/spectraxgk/objectives/stellarator_tables.py
src/spectraxgk/objectives/vmec_boozer.py
src/spectraxgk/objectives/vmec_boozer_context.py
src/spectraxgk/objectives/vmec_boozer_fd.py
src/spectraxgk/objectives/vmec_boozer_gradients.py
src/spectraxgk/objectives/vmec_boozer_line_search.py
src/spectraxgk/objectives/vmec_state.py
src/spectraxgk/objectives/vmec_transport.py
src/spectraxgk/objectives/vmec_transport_branch.py
src/spectraxgk/objectives/vmec_transport_config.py
src/spectraxgk/objectives/vmec_transport_gradient.py
src/spectraxgk/objectives/vmec_transport_line_search.py
src/spectraxgk/objectives/vmec_transport_tables.py
src/spectraxgk/objectives/zonal.py
src/spectraxgk/objectives/zonal_records.py
src/spectraxgk/operators/__init__.py
src/spectraxgk/operators/linear/__init__.py
src/spectraxgk/operators/linear/cache.py
src/spectraxgk/operators/linear/cache_arrays.py
src/spectraxgk/operators/linear/cache_builder.py
src/spectraxgk/operators/linear/cache_model.py
src/spectraxgk/operators/linear/linked.py
src/spectraxgk/operators/linear/moments.py
src/spectraxgk/operators/linear/params.py
src/spectraxgk/operators/linear/rhs.py
src/spectraxgk/operators/linear/streaming.py
src/spectraxgk/operators/nonlinear/__init__.py
src/spectraxgk/operators/nonlinear/collisions.py
src/spectraxgk/operators/nonlinear/device_z.py
src/spectraxgk/operators/nonlinear/device_z_reports.py
src/spectraxgk/operators/nonlinear/diagnostic_state.py
src/spectraxgk/operators/nonlinear/diagnostics.py
src/spectraxgk/operators/nonlinear/domain_decomposition.py
src/spectraxgk/operators/nonlinear/parallel.py
src/spectraxgk/operators/nonlinear/parallel_contracts_domain.py
src/spectraxgk/operators/nonlinear/parallel_contracts_spectral.py
src/spectraxgk/operators/nonlinear/parallel_contracts_strategy.py
src/spectraxgk/operators/nonlinear/policies.py
src/spectraxgk/operators/nonlinear/projection.py
src/spectraxgk/operators/nonlinear/rhs.py
src/spectraxgk/operators/nonlinear/spectral_brackets.py
src/spectraxgk/operators/nonlinear/spectral_core.py
src/spectraxgk/operators/nonlinear/spectral_identity_integrator.py
src/spectraxgk/operators/nonlinear/spectral_identity_reports.py
src/spectraxgk/operators/nonlinear/spectral_identity_rhs.py
src/spectraxgk/operators/nonlinear/spectral_layout.py
src/spectraxgk/operators/nonlinear/spectral_state.py
src/spectraxgk/operators/nonlinear/spectral_tolerances.py
src/spectraxgk/operators/nonlinear/spectral_work_models.py
src/spectraxgk/parallel/__init__.py
src/spectraxgk/parallel/batch.py
src/spectraxgk/parallel/decomposition.py
src/spectraxgk/parallel/identity.py
src/spectraxgk/parallel/independent.py
src/spectraxgk/parallel/integrators.py
src/spectraxgk/parallel/state.py
src/spectraxgk/parallel/velocity.py
src/spectraxgk/parallel/velocity_drive.py
src/spectraxgk/parallel/velocity_hermite.py
src/spectraxgk/parallel/velocity_plan.py
src/spectraxgk/parallel/velocity_streaming.py
src/spectraxgk/solvers/__init__.py
src/spectraxgk/solvers/linear/__init__.py
src/spectraxgk/solvers/linear/eigen_operator.py
src/spectraxgk/solvers/linear/eigen_policy.py
src/spectraxgk/solvers/linear/eigen_preconditioners.py
src/spectraxgk/solvers/linear/eigen_selection.py
src/spectraxgk/solvers/linear/implicit.py
src/spectraxgk/solvers/linear/integrator_diagnostics.py
src/spectraxgk/solvers/linear/integrators.py
src/spectraxgk/solvers/linear/krylov.py
src/spectraxgk/solvers/linear/krylov_algorithms.py
src/spectraxgk/solvers/linear/parallel.py
src/spectraxgk/solvers/linear/parallel_common.py
src/spectraxgk/solvers/linear/parallel_electrostatic.py
src/spectraxgk/solvers/linear/parallel_streaming.py
src/spectraxgk/solvers/nonlinear/__init__.py
src/spectraxgk/solvers/nonlinear/diagnostic_integration.py
src/spectraxgk/solvers/nonlinear/diagnostics.py
src/spectraxgk/solvers/nonlinear/explicit.py
src/spectraxgk/solvers/nonlinear/imex.py
src/spectraxgk/solvers/nonlinear/imex_diagnostics.py
src/spectraxgk/solvers/nonlinear/state_integration.py
src/spectraxgk/solvers/time/__init__.py
src/spectraxgk/solvers/time/diffrax.py
src/spectraxgk/solvers/time/diffrax_core.py
src/spectraxgk/solvers/time/diffrax_linear.py
src/spectraxgk/solvers/time/diffrax_nonlinear.py
src/spectraxgk/solvers/time/diffrax_streaming.py
src/spectraxgk/solvers/time/explicit.py
src/spectraxgk/solvers/time/explicit_cfl.py
src/spectraxgk/solvers/time/explicit_diagnostics.py
src/spectraxgk/solvers/time/explicit_progress.py
src/spectraxgk/solvers/time/explicit_steps.py
src/spectraxgk/solvers/time/runners.py
src/spectraxgk/terms/__init__.py
src/spectraxgk/terms/assembly.py
src/spectraxgk/terms/assembly_core.py
src/spectraxgk/terms/assembly_diagnostics.py
src/spectraxgk/terms/assembly_fields.py
src/spectraxgk/terms/assembly_helpers.py
src/spectraxgk/terms/brackets.py
src/spectraxgk/terms/config.py
src/spectraxgk/terms/fields.py
src/spectraxgk/terms/gyroaveraging.py
src/spectraxgk/terms/integrators.py
src/spectraxgk/terms/linear_dissipation.py
src/spectraxgk/terms/linear_terms.py
src/spectraxgk/terms/nonlinear.py
src/spectraxgk/terms/operators.py
src/spectraxgk/terms/validation.py
src/spectraxgk/terms/reduced/__init__.py
src/spectraxgk/terms/reduced/cetg_integrator.py
src/spectraxgk/terms/reduced/cetg_model.py
src/spectraxgk/terms/reduced/cetg_rhs.py
src/spectraxgk/terms/reduced/cetg_state.py
src/spectraxgk/utils/__init__.py
src/spectraxgk/utils/callbacks.py
src/spectraxgk/validation/__init__.py
src/spectraxgk/validation/autodiff.py
src/spectraxgk/validation/autodiff_finite_difference.py
src/spectraxgk/validation/external_holdout.py
src/spectraxgk/validation/gate_reports.py
src/spectraxgk/validation/gate_types.py
src/spectraxgk/validation/gates.py
src/spectraxgk/validation/zonal.py
src/spectraxgk/validation/benchmarks/__init__.py
src/spectraxgk/validation/benchmarks/batching.py
src/spectraxgk/validation/benchmarks/case_configs.py
src/spectraxgk/validation/benchmarks/cyclone_linear.py
src/spectraxgk/validation/benchmarks/cyclone_linear_paths.py
src/spectraxgk/validation/benchmarks/cyclone_scan.py
src/spectraxgk/validation/benchmarks/cyclone_scan_branches.py
src/spectraxgk/validation/benchmarks/cyclone_scan_explicit.py
src/spectraxgk/validation/benchmarks/cyclone_scan_krylov.py
src/spectraxgk/validation/benchmarks/cyclone_scan_seed.py
src/spectraxgk/validation/benchmarks/defaults.py
src/spectraxgk/validation/benchmarks/etg_linear.py
src/spectraxgk/validation/benchmarks/etg_scan.py
src/spectraxgk/validation/benchmarks/etg_scan_paths.py
src/spectraxgk/validation/benchmarks/fit_signals.py
src/spectraxgk/validation/benchmarks/harness.py
src/spectraxgk/validation/benchmarks/harness_eigenfunctions.py
src/spectraxgk/validation/benchmarks/harness_metrics.py
src/spectraxgk/validation/benchmarks/harness_scan.py
src/spectraxgk/validation/benchmarks/harness_timeseries.py
src/spectraxgk/validation/benchmarks/harness_zonal_metrics.py
src/spectraxgk/validation/benchmarks/initialization.py
src/spectraxgk/validation/benchmarks/kbm_beta.py
src/spectraxgk/validation/benchmarks/kbm_beta_solver_paths.py
src/spectraxgk/validation/benchmarks/kbm_linear.py
src/spectraxgk/validation/benchmarks/kbm_linear_paths.py
src/spectraxgk/validation/benchmarks/kbm_scan.py
src/spectraxgk/validation/benchmarks/kinetic_linear.py
src/spectraxgk/validation/benchmarks/kinetic_scan.py
src/spectraxgk/validation/benchmarks/reference.py
src/spectraxgk/validation/benchmarks/scan.py
src/spectraxgk/validation/benchmarks/secondary.py
src/spectraxgk/validation/benchmarks/solver_policy.py
src/spectraxgk/validation/benchmarks/species.py
src/spectraxgk/validation/benchmarks/tem.py
src/spectraxgk/validation/benchmarks/tem_paths.py
src/spectraxgk/validation/nonlinear_gradient/__init__.py
src/spectraxgk/validation/nonlinear_gradient/evidence.py
src/spectraxgk/validation/nonlinear_gradient/evidence_brackets.py
src/spectraxgk/validation/nonlinear_gradient/evidence_classification.py
src/spectraxgk/validation/nonlinear_gradient/evidence_core.py
src/spectraxgk/validation/nonlinear_gradient/evidence_fd.py
src/spectraxgk/validation/nonlinear_gradient/evidence_gap.py
src/spectraxgk/validation/nonlinear_gradient/evidence_scoring.py
src/spectraxgk/validation/nonlinear_gradient/evidence_screening.py
src/spectraxgk/validation/nonlinear_gradient/evidence_windows.py
src/spectraxgk/validation/nonlinear_gradient/followup_candidate.py
src/spectraxgk/validation/nonlinear_gradient/followup_composite.py
src/spectraxgk/validation/nonlinear_gradient/followup_core.py
src/spectraxgk/validation/nonlinear_gradient/followup_plan.py
src/spectraxgk/validation/nonlinear_gradient/followup_ql_seed.py
src/spectraxgk/validation/nonlinear_gradient/followup_state_runbook.py
src/spectraxgk/validation/nonlinear_gradient/followup_variance.py
src/spectraxgk/validation/nonlinear_transport/__init__.py
src/spectraxgk/validation/nonlinear_transport/optimization_guard.py
src/spectraxgk/validation/nonlinear_transport/optimization_policy.py
src/spectraxgk/validation/nonlinear_transport/optimization_reports.py
src/spectraxgk/validation/nonlinear_transport/replicate_diagnostics.py
src/spectraxgk/validation/nonlinear_transport/replicate_followup.py
src/spectraxgk/validation/quasilinear/__init__.py
src/spectraxgk/validation/quasilinear/calibration_core.py
src/spectraxgk/validation/quasilinear/calibration_io.py
src/spectraxgk/validation/quasilinear/calibration_spectrum.py
src/spectraxgk/validation/quasilinear/holdout_admission.py
src/spectraxgk/validation/quasilinear/model_selection.py
src/spectraxgk/validation/quasilinear/model_selection_inputs.py
src/spectraxgk/validation/quasilinear/window_config.py
src/spectraxgk/validation/quasilinear/window_ensemble.py
src/spectraxgk/validation/quasilinear/window_io.py
src/spectraxgk/validation/quasilinear/window_promotion.py
src/spectraxgk/validation/quasilinear/window_statistics.py
src/spectraxgk/validation/stellarator/__init__.py
src/spectraxgk/validation/stellarator/candidate_gate.py
src/spectraxgk/validation/stellarator/transport_audit.py
src/spectraxgk/validation/stellarator/transport_campaign.py
src/spectraxgk/validation/stellarator/transport_landscape.py
src/spectraxgk/validation/stellarator/transport_policies.py
src/spectraxgk/validation/stellarator/transport_prelaunch.py
src/spectraxgk/validation/stellarator/transport_samples.py
src/spectraxgk/validation/stellarator/transport_selection.py
src/spectraxgk/workflows/__init__.py
src/spectraxgk/workflows/cases.py
src/spectraxgk/workflows/demo.py
src/spectraxgk/workflows/linear.py
src/spectraxgk/workflows/named_cases.py
src/spectraxgk/workflows/nonlinear.py
src/spectraxgk/workflows/reduced_models.py
src/spectraxgk/workflows/runtime/__init__.py
src/spectraxgk/workflows/runtime/artifacts.py
src/spectraxgk/workflows/runtime/chunks.py
src/spectraxgk/workflows/runtime/command_artifacts.py
src/spectraxgk/workflows/runtime/command_options.py
src/spectraxgk/workflows/runtime/commands.py
src/spectraxgk/workflows/runtime/config.py
src/spectraxgk/workflows/runtime/diagnostic_arrays.py
src/spectraxgk/workflows/runtime/diagnostics.py
src/spectraxgk/workflows/runtime/execution.py
src/spectraxgk/workflows/runtime/initial_conditions.py
src/spectraxgk/workflows/runtime/initial_phi.py
src/spectraxgk/workflows/runtime/orchestration.py
src/spectraxgk/workflows/runtime/orchestration_artifacts.py
src/spectraxgk/workflows/runtime/orchestration_progress.py
src/spectraxgk/workflows/runtime/orchestration_scan.py
src/spectraxgk/workflows/runtime/policies.py
src/spectraxgk/workflows/runtime/results.py
src/spectraxgk/workflows/runtime/startup.py
src/spectraxgk/workflows/runtime/toml.py
tests/test_analysis.py
tests/test_audit_repository_size.py
tests/test_auto_fit_signal.py
tests/test_autodiff_demo.py
tests/test_autodiff_twomode_demo.py
tests/test_autodiff_validation.py
tests/test_basis.py
tests/test_benchmark_results_manifest.py
tests/test_benchmark_runtime_memory.py
tests/test_benchmark_scan.py
tests/test_benchmarking.py
tests/test_benchmarks.py
tests/test_benchmarks_helpers.py
tests/test_benchmarks_runner_branches.py
tests/test_build_baseline_optimized_nonlinear_audit.py
tests/test_build_external_vmec_replicate_ensemble.py
tests/test_build_manuscript_readiness_status.py
tests/test_build_matched_nonlinear_transport_comparison.py
tests/test_build_nonlinear_campaign_admission_report.py
tests/test_build_nonlinear_landscape_admission_report.py
tests/test_build_nonlinear_transport_horizon_audit.py
tests/test_build_nonlinear_window_fd_audit.py
tests/test_build_open_research_lane_status.py
tests/test_build_parallelization_completion_status.py
tests/test_build_pre_manuscript_closure_runbook.py
tests/test_build_pre_manuscript_closure_status.py
tests/test_build_qa_itg_optimization_readme_panel.py
tests/test_build_qa_optimizer_strategy_report.py
tests/test_build_qi_branch_refinement_gate.py
tests/test_build_quasilinear_error_anatomy.py
tests/test_build_quasilinear_holdout_gap_report.py
tests/test_build_reduced_nonlinear_audit_prelaunch_report.py
tests/test_build_technical_release_status.py
tests/test_build_tem_branch_parity_audit.py
tests/test_build_transport_audit_redesign_report.py
tests/test_build_vmec_boozer_aggregate_alpha_holdout_gate.py
tests/test_build_vmec_boozer_aggregate_line_search_comparison.py
tests/test_build_vmec_boozer_aggregate_line_search_gate.py
tests/test_build_vmec_boozer_aggregate_objective_gate.py
tests/test_build_vmec_boozer_aggregate_surface_holdout_gate.py
tests/test_build_vmec_boozer_gradient_holdout_matrix.py
tests/test_build_vmec_boozer_multi_point_objective_gate.py
tests/test_build_vmec_boozer_nonlinear_window_fd_audit.py
tests/test_build_vmec_boozer_nonlinear_window_gradient_gate.py
tests/test_build_vmec_boozer_parity_matrix.py
tests/test_build_vmec_boozer_production_holdout_artifact.py
tests/test_build_vmec_boozer_second_equilibrium_aggregate_gate.py
tests/test_build_vmec_jax_boundary_chain_collection.py
tests/test_build_vmec_jax_qa_full_sweep_panel.py
tests/test_build_vmec_jax_qa_transport_candidate_comparison.py
tests/test_build_vmec_jax_qa_transport_optimization_status.py
tests/test_build_vmec_jax_transport_gradient_diagnostic.py
tests/test_build_vmec_optimization_candidate_screen_gate.py
tests/test_build_vmec_state_control_bracket_sweep_status.py
tests/test_build_vmec_state_to_input_mapping_response.py
tests/test_build_w7x_tem_extension_status.py
tests/test_build_zonal_flow_objective_gate.py
tests/test_callbacks.py
tests/test_cetg.py
tests/test_check_external_vmec_high_grid_admission.py
tests/test_check_nonlinear_runtime_outputs.py
tests/test_check_nonlinear_window_ensemble.py
tests/test_check_nonlinear_window_ensemble_readiness.py
tests/test_check_overdetermined_nonlinear_gradient_campaign.py
tests/test_check_quasilinear_calibration_inputs.py
tests/test_check_release_artifact_manifest.py
tests/test_check_release_readiness.py
tests/test_check_release_version.py
tests/test_check_repository_size_manifest.py
tests/test_check_vmec_boozer_aggregate_holdout_gate.py
tests/test_check_vmec_boozer_differentiability_claim.py
tests/test_check_vmec_boozer_reduced_portfolio_guard.py
tests/test_cli.py
tests/test_compact_replicate_ensemble_bundle.py
tests/test_compare_gx_fieldsolve_dump.py
tests/test_compare_gx_imported_bigfield_linear.py
tests/test_compare_gx_imported_growth_dump.py
tests/test_compare_gx_imported_linear.py
tests/test_compare_gx_imported_startup.py
tests/test_compare_gx_imported_window.py
tests/test_compare_gx_kbm.py
tests/test_compare_gx_nonlinear.py
tests/test_compare_gx_nonlinear_diagnostics.py
tests/test_compare_gx_nonlinear_rk4_stage.py
tests/test_compare_gx_nonlinear_terms.py
tests/test_compare_gx_rhs_terms.py
tests/test_compare_gx_rk4_stage.py
tests/test_compare_gx_runtime_diag_state.py
tests/test_compare_gx_runtime_startup.py
tests/test_compare_gx_runtime_window.py
tests/test_compare_gx_secondary.py
tests/test_compare_gx_startup.py
tests/test_compare_w7x_zonal_reference.py
tests/test_compress_docs_previews.py
tests/test_compress_release_previews.py
tests/test_config.py
tests/test_core_contracts.py
tests/test_derive_imported_linear_lastvalue_table.py
tests/test_device_parity_gate.py
tests/test_differentiable_geometry.py
tests/test_differentiable_geometry_bridge.py
tests/test_differentiable_refactor_manifest.py
tests/test_diffrax_integrators.py
tests/test_diffrax_integrators_core.py
tests/test_digitize_w7x_zonal_reference.py
tests/test_evaluate_vmec_jax_spectrax_transport_metric.py
tests/test_example_smoke.py
tests/test_explicit_time_integrators_lowlevel.py
tests/test_external_holdout_plan.py
tests/test_fast_smoke.py
tests/test_full_operator.py
tests/test_gate_laguerre_nonlinear_modes.py
tests/test_gate_linear_rhs_zero_norm_state_window.py
tests/test_generate_electrostatic_diamagnetic_gate.py
tests/test_generate_electrostatic_drift_gate.py
tests/test_generate_electrostatic_field_reduce_gate.py
tests/test_generate_hermite_exchange_gate.py
tests/test_generate_hermite_streaming_ladder_gate.py
tests/test_generate_kbm_branch_gate_summary.py
tests/test_generate_kbm_reference_overlay.py
tests/test_generate_linear_rhs_electrostatic_slices_gate.py
tests/test_generate_linear_rhs_streaming_electrostatic_gate.py
tests/test_generate_linear_rhs_streaming_gate.py
tests/test_generate_logical_cpu_parallel_scan_gate.py
tests/test_generate_miller_zonal_response_pilot.py
tests/test_generate_nonlinear_sharding_production_gate.py
tests/test_generate_observed_order_gate.py
tests/test_generate_parallel_ky_scan_gate.py
tests/test_generate_periodic_streaming_microkernel_gate.py
tests/test_generate_quasilinear_runtime_parallel_gate.py
tests/test_generate_velocity_field_reduce_gate.py
tests/test_generate_w7x_reference_overlay.py
tests/test_generate_w7x_zonal_response_panel.py
tests/test_geometry.py
tests/test_geometry_backend_kernels.py
tests/test_geometry_backend_miller_internal.py
tests/test_geometry_backend_vmec_helpers.py
tests/test_geometry_miller_lowlevel.py
tests/test_geometry_standalone.py
tests/test_grids.py
tests/test_gx_consistency.py
tests/test_gyroaverage.py
tests/test_hyperdiffusion.py
tests/test_independent_ky_scaling_artifacts.py
tests/test_io.py
tests/test_linear.py
tests/test_linear_helpers_extra.py
tests/test_linear_krylov_core.py
tests/test_linear_moments_invariants.py
tests/test_linear_operator_packages.py
tests/test_linear_parallel_dispatch.py
tests/test_make_benchmark_atlas.py
tests/test_make_cyclone_assets.py
tests/test_make_gx_cyclone_kbm_panel.py
tests/test_make_gx_publication_panel.py
tests/test_make_gx_summary_panel.py
tests/test_make_validation_gate_index.py
tests/test_miller_eik.py
tests/test_nonlinear.py
tests/test_nonlinear_diagnostic_state.py
tests/test_nonlinear_domain_parallel.py
tests/test_nonlinear_exb.py
tests/test_nonlinear_explicit_step.py
tests/test_nonlinear_gradient_evidence.py
tests/test_nonlinear_gradient_followup.py
tests/test_nonlinear_gradient_ql_seed_screen.py
tests/test_nonlinear_gradient_state_control_runbook.py
tests/test_nonlinear_helpers_extra.py
tests/test_nonlinear_imex.py
tests/test_nonlinear_operator_packages.py
tests/test_nonlinear_parallel.py
tests/test_nonlinear_replicate_diagnostics.py
tests/test_nonlinear_replicate_followup.py
tests/test_nonlinear_rhs.py
tests/test_nonlinear_sharding_artifacts.py
tests/test_nonlinear_spectral_communication_gate.py
tests/test_nonlinear_transport_optimization.py
tests/test_normalization.py
tests/test_operators.py
tests/test_package_architecture_manifest.py
tests/test_parallel.py
tests/test_parallel_artifact_contracts.py
tests/test_parallel_decomposition.py
tests/test_parallelization_examples.py
tests/test_performance_optimization_manifest.py
tests/test_plot_external_vmec_nonlinear_convergence_gate.py
tests/test_plot_external_vmec_time_horizon_gate.py
tests/test_plot_independent_ky_scan_scaling.py
tests/test_plot_nonlinear_feasibility_pilot.py
tests/test_plot_nonlinear_rhs_profile.py
tests/test_plot_nonlinear_sharding_strong_scaling.py
tests/test_plot_nonlinear_window_statistics.py
tests/test_plot_quasilinear_calibration.py
tests/test_plot_quasilinear_candidate_regularization_sweep.py
tests/test_plot_quasilinear_candidate_uncertainty.py
tests/test_plot_quasilinear_dataset_sufficiency.py
tests/test_plot_quasilinear_model_selection_status.py
tests/test_plot_quasilinear_saturation_rule_sweep.py
tests/test_plot_quasilinear_screening_skill.py
tests/test_plot_quasilinear_shape_aware_saturation.py
tests/test_plot_quasilinear_spectrum.py
tests/test_plot_quasilinear_spectrum_shape_gate.py
tests/test_plot_quasilinear_stellarator_usefulness.py
tests/test_plot_quasilinear_uq_ensemble_scaling.py
tests/test_plot_stellarator_optimization_uq.py
tests/test_plot_vmec_jax_equilibrium_inventory.py
tests/test_plot_w7x_exact_state_audit.py
tests/test_plot_w7x_fluctuation_spectrum_panel.py
tests/test_plot_w7x_zonal_closure_ladder.py
tests/test_plot_w7x_zonal_contract_audit.py
tests/test_plot_w7x_zonal_moment_tail_audit.py
tests/test_plot_w7x_zonal_recurrence_sweep.py
tests/test_plot_w7x_zonal_state_convention_audit.py
tests/test_plot_zonal_flow_response_from_output.py
tests/test_plotting.py
tests/test_postprocess_nonlinear_gradient_control_mean_campaign.py
tests/test_postprocess_overdetermined_nonlinear_gradient_campaign.py
tests/test_prepare_external_vmec_holdout_from_screen.py
tests/test_probe_gx_kbm_extractors.py
tests/test_profile_device_z_pencil_transport_window.py
tests/test_profile_full_linear_rhs_trace.py
tests/test_profile_full_nonlinear_rhs_trace.py
tests/test_profile_independent_ky_scan_scaling.py
tests/test_profile_linear_cache_build.py
tests/test_profile_linear_rhs_parallel_slices.py
tests/test_profile_linear_rhs_parallel_slices_sweep.py
tests/test_profile_linear_rhs_terms.py
tests/test_profile_nonlinear_sharding.py
tests/test_profile_nonlinear_sharding_sweep.py
tests/test_profile_nonlinear_spectral_domain_routing.py
tests/test_profile_quasilinear_uq_ensemble_scaling.py
tests/test_profile_runtime_startup.py
tests/test_profiler_options.py
tests/test_public_api.py
tests/test_qa_low_turbulence.py
tests/test_quasilinear.py
tests/test_quasilinear_calibration.py
tests/test_quasilinear_implicit_sensitivity_demo.py
tests/test_quasilinear_model_selection.py
tests/test_quasilinear_model_selection_extra.py
tests/test_quasilinear_promotion_guardrails.py
tests/test_quasilinear_uq_ensemble_scaling_artifacts.py
tests/test_quasilinear_window.py
tests/test_reduced_model_contracts.py
tests/test_refactor_coverage_manifest.py
tests/test_release_scope_docs.py
tests/test_restart.py
tests/test_restart_gate.py
tests/test_run_benchmark_refresh.py
tests/test_run_device_parity_gate.py
tests/test_run_exact_state_audit.py
tests/test_run_imported_linear_targeted_audit.py
tests/test_run_kbm_lowky_extractor_audit.py
tests/test_run_nonlinear_gradient_direct_campaign.py
tests/test_run_nonlinear_gradient_manifest_postprocess.py
tests/test_run_overdetermined_nonlinear_gradient_campaign.py
tests/test_run_restart_parity_gate.py
tests/test_run_tests_fast.py
tests/test_run_vmec_jax_guarded_transport_ladder.py
tests/test_run_vmec_roundtrip_gate.py
tests/test_run_wide_coverage_gate.py
tests/test_runners.py
tests/test_runtime_artifacts.py
tests/test_runtime_chunks.py
tests/test_runtime_config.py
tests/test_runtime_diagnostics.py
tests/test_runtime_helpers.py
tests/test_runtime_policies.py
tests/test_runtime_results.py
tests/test_runtime_runner.py
tests/test_secondary.py
tests/test_sharded_integrators.py
tests/test_sharding.py
tests/test_solver_gradient_gate_modules.py
tests/test_solver_objective_gradients.py
tests/test_species.py
tests/test_stellarator_objective_portfolio.py
tests/test_stellarator_optimization.py
tests/test_summarize_nonlinear_replicate_spread.py
tests/test_terms_assembly.py
tests/test_terms_fields.py
tests/test_terms_integrators.py
tests/test_terms_operators.py
tests/test_terms_validation.py
tests/test_validation_coverage_manifest.py
tests/test_validation_gates.py
tests/test_velocity_sharding.py
tests/test_vmec_boundary_transport_landscape.py
tests/test_vmec_eik.py
tests/test_vmec_example_inventory.py
tests/test_vmec_jax_boundary_chain.py
tests/test_vmec_jax_candidate_gate.py
tests/test_vmec_jax_qa_exact_scripts.py
tests/test_vmec_jax_qa_optimization_driver.py
tests/test_vmec_jax_transport_admission.py
tests/test_vmec_jax_transport_gradient.py
tests/test_vmec_jax_transport_line_search.py
tests/test_vmec_jax_transport_objective.py
tests/test_vmec_roundtrip_gate.py
tests/test_write_external_vmec_holdout_configs.py
tests/test_write_nonlinear_replicate_followup_campaign.py
tests/test_write_optimized_equilibrium_transport_configs.py
tests/test_write_overdetermined_nonlinear_gradient_campaign.py
tests/test_write_vmec_asymmetric_state_to_input_mapping_campaign.py
tests/test_write_vmec_boundary_perturbation_inputs.py
tests/test_write_vmec_boundary_profile_perturbation_inputs.py
tests/test_write_vmec_jax_optimizer_comparison_manifest.py
tests/test_write_vmec_jax_projected_transport_line_search_inputs.py
tests/test_write_vmec_jax_spsa_candidate_campaign.py
tests/test_write_vmec_qa_t1500_postprocess_manifest.py
tests/test_write_vmec_state_control_short_bracket_launch.py
tests/test_write_vmec_state_to_input_mapping_campaign.py
tests/test_write_w7x_zonal_closure_sweep.py
tests/test_zonal_objective.py
tests/test_zonal_validation.py