Index

_

_apply_collision_split() (in module spectraxgk.nonlinear_helpers) _as_species_array() (in module spectraxgk.linear_params) _build_end_damping_profile_array() (in module spectraxgk.linear_cache) _build_gyroaverage_cache_arrays() (in module spectraxgk.linear_cache) _build_linked_end_damping_profile() (in module spectraxgk.linear_linked) _build_linked_fft_maps() (in module spectraxgk.linear_linked) _build_low_rank_moment_cache_arrays() (in module spectraxgk.linear_cache) _collision_damping() (in module spectraxgk.nonlinear_helpers) _electrostatic_streaming_field_rhs() (in module spectraxgk.linear_parallel) _gx_nonlinear_omega_components() (in module spectraxgk.nonlinear_helpers)

_gx_omega_mode_mask() (in module spectraxgk.nonlinear_helpers) _infer_runtime_nonlinear_steps() (in module spectraxgk.runtime_policies) _linear_rhs_electrostatic_slices_velocity_sharded_fused() (in module spectraxgk.linear_parallel) _make_fixed_mode_projector() (in module spectraxgk.nonlinear_helpers) _make_hermitian_projector() (in module spectraxgk.nonlinear_helpers) _parallel_requests_combined_ky_scan() (in module spectraxgk.runtime_policies) _resolve_parallel_devices() (in module spectraxgk.linear_parallel) _runtime_external_phi() (in module spectraxgk.runtime_policies) _runtime_independent_parallel_plan() (in module spectraxgk.runtime_policies) _sample_indices_with_final() (in module spectraxgk.nonlinear_diagnostics) _streaming_electrostatic_from_phi_velocity_sharded() (in module spectraxgk.linear_parallel)

A

aggregate_objective_portfolio() (in module spectraxgk.stellarator_objective_portfolio) apply_auto_fit_scan_policy() (in module spectraxgk.benchmark_scan) apply_diagnostic_normalization() (in module spectraxgk.normalization) apply_geometry_grid_defaults() (in module spectraxgk.geometry) apply_gx_geometry_grid_defaults() (in module spectraxgk.geometry)

apply_heat_flux_scale() (in module spectraxgk.quasilinear_calibration) apply_hermite_v() (in module spectraxgk.linear_moments) apply_hermite_v2() (in module spectraxgk.linear_moments) apply_laguerre_x() (in module spectraxgk.linear_moments) autodiff_finite_difference_report() (in module spectraxgk.autodiff_validation)

B

batch_map() (in module spectraxgk.parallel) batch_map_identity_report() (in module spectraxgk.parallel) bessel_j0() (in module spectraxgk.gyroaverage) bessel_j1() (in module spectraxgk.gyroaverage) bgrad() (spectraxgk.geometry.SAlphaGeometry method) bmag() (spectraxgk.geometry.SAlphaGeometry method) booz_xform_flux_tube_mapping_from_inputs() (in module spectraxgk.geometry.differentiable) booz_xform_flux_tube_sensitivity_report() (in module spectraxgk.geometry.differentiable) booz_xform_spectral_sensitivity_report() (in module spectraxgk.geometry.differentiable) boundary_chain_accepted_parameter_indices() (in module spectraxgk.vmec_jax_transport_line_search) boundary_chain_summary_from_probe() (in module spectraxgk.vmec_jax_boundary_chain) boundary_spec_record() (in module spectraxgk.vmec_jax_transport_gradient) branch_continuity_gate_report() (in module spectraxgk.validation_gates) branch_continuity_metrics() (in module spectraxgk.benchmarking) BranchContinuationMetrics (class in spectraxgk.validation_gates) build_authoritative_wout_candidate_gate() (in module spectraxgk.vmec_jax_candidate_gate) build_boundary_chain_collection_summary() (in module spectraxgk.vmec_jax_boundary_chain) build_boundary_chain_summary() (in module spectraxgk.vmec_jax_boundary_chain) build_boundary_transport_gradient_report() (in module spectraxgk.vmec_jax_transport_gradient) build_diagnostic_nonlinear_domain_decomposition() (in module spectraxgk.parallel_decomposition)

build_external_holdout_runbook() (in module spectraxgk.external_holdout_plan) build_flux_tube_geometry() (in module spectraxgk.geometry) build_H() (in module spectraxgk.linear_moments) build_independent_portfolio_decomposition() (in module spectraxgk.parallel_decomposition) build_linear_cache() (in module spectraxgk.linear_cache) build_nonlinear_audit_redesign_report() (in module spectraxgk.vmec_jax_transport_admission) build_nonlinear_campaign_admission_report() (in module spectraxgk.vmec_jax_transport_admission) build_nonlinear_imex_operator() (in module spectraxgk.nonlinear_helpers) build_nonlinear_landscape_admission_report() (in module spectraxgk.vmec_jax_transport_admission) build_quasilinear_model_selection_status() (in module spectraxgk.quasilinear_model_selection) build_quasilinear_model_selection_status_from_paths() (in module spectraxgk.quasilinear_model_selection) build_reduced_nonlinear_audit_prelaunch_report() (in module spectraxgk.vmec_jax_transport_admission) build_runtime_geometry() (in module spectraxgk.runtime) build_runtime_linear_params() (in module spectraxgk.runtime) build_runtime_linear_terms() (in module spectraxgk.runtime) build_runtime_progress_message() (in module spectraxgk.runtime_orchestration) build_runtime_term_config() (in module spectraxgk.runtime) build_solved_vmec_candidate_gate() (in module spectraxgk.vmec_jax_candidate_gate) build_transport_admission_report() (in module spectraxgk.vmec_jax_transport_admission) build_velocity_sharding_plan() (in module spectraxgk.velocity_sharding) build_wout_reproducibility_gate() (in module spectraxgk.vmec_jax_candidate_gate)

C

calibration_point_from_nonlinear_window_summary() (in module spectraxgk.quasilinear_calibration) calibration_point_from_spectrum_and_nonlinear_window() (in module spectraxgk.quasilinear_calibration) candidate_transport_metric() (in module spectraxgk.vmec_jax_transport_admission) case (spectraxgk.normalization.NormalizationContract attribute) central_finite_difference_jacobian() (in module spectraxgk.autodiff_validation) collision_damping() (in module spectraxgk.linear_cache) compare_cyclone_to_reference() (in module spectraxgk.benchmarks) compare_eigenfunctions() (in module spectraxgk.benchmarking) compute_b() (in module spectraxgk.linear_moments) compute_quasilinear_from_linear_state() (in module spectraxgk.quasilinear)

covariance_diagnostics() (in module spectraxgk.autodiff_validation) curvature_gradb_drift_reference() (in module spectraxgk.velocity_sharding) curvature_gradb_drift_shard_map() (in module spectraxgk.velocity_sharding) cyclone_comparison_figure() (in module spectraxgk.plotting) cyclone_reference_figure() (in module spectraxgk.plotting) CycloneBaseCase (class in spectraxgk.config) CycloneComparison (class in spectraxgk.benchmarks) CycloneReference (class in spectraxgk.benchmarks) CycloneRunResult (class in spectraxgk.benchmarks) CycloneScanResult (class in spectraxgk.benchmarks)

D

DecompositionContract (class in spectraxgk.parallel_decomposition) default_qa_low_turbulence_initial_params() (in module spectraxgk.qa_low_turbulence) density_moment() (in module spectraxgk.analysis) diagnostic_nonlinear_partition (spectraxgk.parallel_decomposition.DecompositionContract property) diagnostic_norm_default (spectraxgk.normalization.NormalizationContract attribute) DiagnosticTimeSeries (class in spectraxgk.validation_gates)

diamagnetic_drive_coeffs() (in module spectraxgk.linear_moments) diamagnetic_drive_reference() (in module spectraxgk.velocity_sharding) diamagnetic_drive_shard_map() (in module spectraxgk.velocity_sharding) discover_differentiable_geometry_backends() (in module spectraxgk.geometry.differentiable) drift_coeffs() (spectraxgk.geometry.SAlphaGeometry method) drift_components() (spectraxgk.geometry.SAlphaGeometry method)

E

effective_boundary() (in module spectraxgk.geometry) effective_kperp2() (in module spectraxgk.quasilinear) eigenfunction_gate_report() (in module spectraxgk.validation_gates) eigenfunction_overlap_summary_figure() (in module spectraxgk.plotting) eigenfunction_reference_overlay_figure() (in module spectraxgk.plotting) EigenfunctionComparisonMetrics (class in spectraxgk.validation_gates) EigenfunctionReferenceBundle (class in spectraxgk.validation_gates) electrostatic_phi_reference() (in module spectraxgk.velocity_sharding) electrostatic_phi_shard_map() (in module spectraxgk.velocity_sharding) enabled (spectraxgk.runtime.RuntimeIndependentParallelPlan property) (spectraxgk.runtime_policies.RuntimeIndependentParallelPlan property) energy_operator() (in module spectraxgk.linear_moments) ensure_flux_tube_geometry_data() (in module spectraxgk.geometry)

estimate_observed_order() (in module spectraxgk.benchmarking) etg_trend_figure() (in module spectraxgk.plotting) ETGBaseCase (class in spectraxgk.config) ETGModelConfig (class in spectraxgk.config) evaluate_boozer_bmag_on_field_line() (in module spectraxgk.geometry.differentiable) evaluate_scalar_gate() (in module spectraxgk.validation_gates) explicit_complex_operator_matrix() (in module spectraxgk.autodiff_validation) explicit_method_default_cfl_fac() (in module spectraxgk.config) external_vmec_family() (in module spectraxgk.external_holdout_plan) ExternalHoldoutScreenRow (class in spectraxgk.external_holdout_plan) extract_eigenfunction() (in module spectraxgk.analysis) extract_mode() (in module spectraxgk.analysis) extract_mode_time_series() (in module spectraxgk.analysis)

F

family (spectraxgk.external_holdout_plan.ExternalHoldoutScreenRow property) final_iota_profiles_from_vmec_result() (in module spectraxgk.vmec_jax_candidate_gate) finite_difference_jacobian() (in module spectraxgk.geometry.differentiable) fit_growth_rate() (in module spectraxgk.analysis) fit_growth_rate_auto() (in module spectraxgk.analysis) fit_growth_rate_auto_fn() (spectraxgk.benchmark_scan.ScanFitWindowPolicy method) fit_growth_rate_auto_with_stats() (in module spectraxgk.analysis) fit_growth_rate_fn() (spectraxgk.benchmark_scan.ScanFitWindowPolicy method)

fit_growth_rate_with_stats() (in module spectraxgk.analysis) fit_signal() (spectraxgk.benchmark_scan.ScanFitWindowPolicy method) fit_train_heat_flux_scale() (in module spectraxgk.quasilinear_calibration) flux_tube_geometry_from_mapping() (in module spectraxgk.geometry.differentiable) flux_tube_geometry_from_vmec_boozer_state() (in module spectraxgk.geometry.differentiable) flux_tube_geometry_observables() (in module spectraxgk.geometry.differentiable) FluxTubeGeometryData (class in spectraxgk.geometry) format_duration() (in module spectraxgk.runtime_orchestration)

G

gamma0() (in module spectraxgk.gyroaverage) gate_report() (in module spectraxgk.validation_gates) gate_report_to_dict() (in module spectraxgk.validation_gates) GateReport (class in spectraxgk.validation_gates) geometry_inverse_design_report() (in module spectraxgk.geometry.differentiable) geometry_observable_names() (in module spectraxgk.geometry.differentiable) geometry_sensitivity_report() (in module spectraxgk.geometry.differentiable) GeometryConfig (class in spectraxgk.config) get_normalization_contract() (in module spectraxgk.normalization) grad_z_periodic() (in module spectraxgk.linear_moments) gradient_scope (spectraxgk.vmec_jax_transport_objective.VMECJAXTransportObjectiveConfig property) gradpar() (spectraxgk.geometry.SAlphaGeometry method) GridConfig (class in spectraxgk.config)

growth_fit_figure() (in module spectraxgk.plotting) growth_rate_heatmap() (in module spectraxgk.plotting) gx_default_cfl_fac() (in module spectraxgk.config) gx_effective_boundary() (in module spectraxgk.geometry) gx_factorial() (in module spectraxgk.gyroaverage) gx_growth_rate_from_omega_series() (in module spectraxgk.analysis) gx_growth_rate_from_phi() (in module spectraxgk.analysis) gx_laguerre_nj() (in module spectraxgk.gyroaverage) gx_laguerre_transform() (in module spectraxgk.gyroaverage) gx_real_fft_kx() (in module spectraxgk.grids) gx_real_fft_ky() (in module spectraxgk.grids) gx_real_fft_mesh() (in module spectraxgk.grids) gx_twist_shift_params() (in module spectraxgk.geometry) gx_zero_shat_enabled() (in module spectraxgk.geometry)

H

hermite_ladder_coeffs() (in module spectraxgk.basis) hermite_neighbor_reference() (in module spectraxgk.velocity_sharding) hermite_neighbor_shard_map() (in module spectraxgk.velocity_sharding) hermite_normed() (in module spectraxgk.basis) hermite_physicists() (in module spectraxgk.basis)

hermite_shift_reference() (in module spectraxgk.velocity_sharding) hermite_shift_shard_map() (in module spectraxgk.velocity_sharding) hermite_streaming() (in module spectraxgk.operators) hermite_streaming_ladder_reference() (in module spectraxgk.velocity_sharding) hermite_streaming_ladder_shard_map() (in module spectraxgk.velocity_sharding) hypercollision_damping() (in module spectraxgk.linear_cache)

I

IMEXLinearOperator (class in spectraxgk.nonlinear_helpers) implicit_eigenpair_observable_sensitivity_report() (in module spectraxgk.autodiff_validation) independent_ensemble_provenance_gate() (in module spectraxgk.parallel) independent_map() (in module spectraxgk.parallel) independent_map_identity_report() (in module spectraxgk.parallel) independent_worker_metadata() (in module spectraxgk.parallel) IndependentEnsembleProvenanceReport (class in spectraxgk.parallel) IndependentMapExecutionError IndependentWorkerMetadata (class in spectraxgk.parallel) indexed_float_value() (in module spectraxgk.benchmark_scan)

indexed_scan_value() (in module spectraxgk.benchmark_scan) infer_triple_dealiased_ny() (in module spectraxgk.benchmarking) InitializationConfig (class in spectraxgk.config) integrate_linear() (in module spectraxgk.linear) integrate_linear_diagnostics() (in module spectraxgk.linear) integrate_linear_sharded() (in module spectraxgk.sharded_integrators) integrate_nonlinear_sharded() (in module spectraxgk.sharded_integrators) integrated_quasilinear_flux_from_spectrum() (in module spectraxgk.quasilinear_calibration) isolated_eigenpair_observable_sensitivity_report() (in module spectraxgk.autodiff_validation) isolated_eigenvalue_sensitivity_report() (in module spectraxgk.autodiff_validation)

J

J() (spectraxgk.vmec_jax_transport_objective.VMECJAXSpectraxTransportObjective method)

J_l_all() (in module spectraxgk.gyroaverage)

K

k_perp2() (spectraxgk.geometry.SAlphaGeometry method) KBMBaseCase (class in spectraxgk.config) KineticElectronBaseCase (class in spectraxgk.config)

KineticElectronModelConfig (class in spectraxgk.config) kx_effective() (spectraxgk.geometry.SAlphaGeometry method) ky_scan_batches() (in module spectraxgk.parallel)

L

laguerre() (in module spectraxgk.basis) late_time_linear_metrics() (in module spectraxgk.benchmarking) late_time_window() (in module spectraxgk.benchmarking) LateTimeLinearMetrics (class in spectraxgk.validation_gates) lenard_bernstein_eigenvalues() (in module spectraxgk.linear_moments) linear_metrics_gate_report() (in module spectraxgk.validation_gates) linear_rhs() (in module spectraxgk.linear) linear_rhs_cached() (in module spectraxgk.linear) linear_rhs_electrostatic_slices_velocity_sharded() (in module spectraxgk.linear_parallel) linear_rhs_parallel_cached() (in module spectraxgk.linear_parallel) linear_rhs_streaming_electrostatic_velocity_sharded() (in module spectraxgk.linear_parallel) linear_rhs_streaming_velocity_sharded() (in module spectraxgk.linear_parallel) linear_runtime_panel_figure() (in module spectraxgk.plotting) linear_terms_to_term_config() (in module spectraxgk.linear_params) linear_validation_figure() (in module spectraxgk.plotting)

linear_validation_multi_reference_figure() (in module spectraxgk.plotting) LinearCache (class in spectraxgk.linear_cache) LinearParams (class in spectraxgk.linear_params) LinearRunResult (class in spectraxgk.benchmarks) LinearScanResult (class in spectraxgk.benchmarks) LinearTerms (class in spectraxgk.linear_params) LinearValidationPanel (class in spectraxgk.plotting) load_cyclone_reference() (in module spectraxgk.benchmarks) load_cyclone_reference_kinetic() (in module spectraxgk.benchmarks) load_diagnostic_time_series() (in module spectraxgk.benchmarking) load_eigenfunction_reference_bundle() (in module spectraxgk.benchmarking) load_etg_reference() (in module spectraxgk.benchmarks) load_gx_geometry_netcdf() (in module spectraxgk.geometry) load_kbm_reference() (in module spectraxgk.benchmarks) load_tem_reference() (in module spectraxgk.benchmarks)

M

metric_coeffs() (spectraxgk.geometry.SAlphaGeometry method) mirror_drift_reference() (in module spectraxgk.velocity_sharding) mirror_drift_shard_map() (in module spectraxgk.velocity_sharding) mixing_length_amplitude2_jax() (in module spectraxgk.quasilinear) ModelConfig (class in spectraxgk.config) ModeSelection (class in spectraxgk.analysis) ModeSelectionBatch (class in spectraxgk.analysis) module spectraxgk.analysis spectraxgk.autodiff_validation spectraxgk.basis spectraxgk.benchmark_defaults spectraxgk.benchmark_scan spectraxgk.benchmarking spectraxgk.benchmarks spectraxgk.config spectraxgk.external_holdout_plan spectraxgk.geometry spectraxgk.geometry.differentiable spectraxgk.grids spectraxgk.gyroaverage spectraxgk.linear spectraxgk.linear_cache spectraxgk.linear_linked spectraxgk.linear_moments spectraxgk.linear_parallel spectraxgk.linear_params spectraxgk.nonlinear_diagnostics spectraxgk.nonlinear_gradient_followup spectraxgk.nonlinear_helpers spectraxgk.nonlinear_replicate_diagnostics spectraxgk.nonlinear_replicate_followup spectraxgk.normalization spectraxgk.operators spectraxgk.parallel spectraxgk.parallel_decomposition spectraxgk.plotting spectraxgk.qa_low_turbulence spectraxgk.quasilinear spectraxgk.quasilinear_calibration spectraxgk.quasilinear_model_selection spectraxgk.runtime spectraxgk.runtime_config spectraxgk.runtime_orchestration spectraxgk.runtime_policies spectraxgk.sharded_integrators spectraxgk.stellarator_objective_portfolio spectraxgk.validation_gates spectraxgk.velocity_sharding spectraxgk.vmec_jax_boundary_chain spectraxgk.vmec_jax_candidate_gate spectraxgk.vmec_jax_transport_admission spectraxgk.vmec_jax_transport_gradient spectraxgk.vmec_jax_transport_line_search spectraxgk.vmec_jax_transport_objective spectraxgk.zonal_objective

MultiReferenceValidationPanel (class in spectraxgk.plotting)

N

n_samples (spectraxgk.stellarator_objective_portfolio.StellaratorObjectivePortfolioContract property) nonlinear_gradient_candidate_design_report() (in module spectraxgk.nonlinear_gradient_followup) nonlinear_gradient_composite_control_report() (in module spectraxgk.nonlinear_gradient_followup) nonlinear_gradient_control_mean_gate() (in module spectraxgk.nonlinear_gradient_followup) nonlinear_gradient_control_variate_campaign_plan() (in module spectraxgk.nonlinear_gradient_followup) nonlinear_gradient_followup_plan() (in module spectraxgk.nonlinear_gradient_followup) nonlinear_gradient_ql_seed_screen_report() (in module spectraxgk.nonlinear_gradient_followup) nonlinear_gradient_state_control_runbook_report() (in module spectraxgk.nonlinear_gradient_followup) nonlinear_gradient_variance_reduction_plan() (in module spectraxgk.nonlinear_gradient_followup) nonlinear_heat_flux_convergence_gate_report() (in module spectraxgk.validation_gates) nonlinear_heat_flux_convergence_metrics() (in module spectraxgk.benchmarking) nonlinear_replicate_followup_plan() (in module spectraxgk.nonlinear_replicate_followup) nonlinear_replicate_spread_report() (in module spectraxgk.nonlinear_replicate_diagnostics) nonlinear_runtime_panel_figure() (in module spectraxgk.plotting) nonlinear_window_gate_report() (in module spectraxgk.validation_gates) NonlinearArtifactPolicy (class in spectraxgk.runtime_orchestration)

NonlinearGradientCandidateDesignConfig (class in spectraxgk.nonlinear_gradient_followup) NonlinearGradientCompositeControlConfig (class in spectraxgk.nonlinear_gradient_followup) NonlinearGradientControlMeanGateConfig (class in spectraxgk.nonlinear_gradient_followup) NonlinearGradientControlVariateCampaignConfig (class in spectraxgk.nonlinear_gradient_followup) NonlinearGradientFollowupConfig (class in spectraxgk.nonlinear_gradient_followup) NonlinearGradientQLSeedScreenConfig (class in spectraxgk.nonlinear_gradient_followup) NonlinearGradientStateControlRunbookConfig (class in spectraxgk.nonlinear_gradient_followup) NonlinearGradientVarianceReductionConfig (class in spectraxgk.nonlinear_gradient_followup) NonlinearHeatFluxConvergenceMetrics (class in spectraxgk.validation_gates) NonlinearReplicateFollowupConfig (class in spectraxgk.nonlinear_replicate_followup) NonlinearReplicateSpreadConfig (class in spectraxgk.nonlinear_replicate_diagnostics) NonlinearWindowMetrics (class in spectraxgk.validation_gates) NormalizationContract (class in spectraxgk.normalization) normalize_eigenfunction() (in module spectraxgk.benchmarking) normalize_fit_signal() (in module spectraxgk.benchmark_scan) normalize_quasilinear_channels() (in module spectraxgk.quasilinear) normalize_solver_key() (in module spectraxgk.benchmark_scan)

O

objective_options() (spectraxgk.vmec_jax_transport_objective.VMECJAXTransportObjectiveConfig method) objective_portfolio_sensitivity_report() (in module spectraxgk.stellarator_objective_portfolio) objective_weights() (spectraxgk.zonal_objective.ZonalFlowObjectiveConfig method) observable_gradient_validation_report() (in module spectraxgk.geometry.differentiable) observed_order_gate_report() (in module spectraxgk.validation_gates)

ObservedOrderMetrics (class in spectraxgk.validation_gates) omega_d() (spectraxgk.geometry.SAlphaGeometry method) omega_d_scale (spectraxgk.normalization.NormalizationContract attribute) omega_star_scale (spectraxgk.normalization.NormalizationContract attribute) optimize_qa_low_turbulence() (in module spectraxgk.qa_low_turbulence)

P

pad_to_multiple() (in module spectraxgk.parallel) parallel_identity_report() (in module spectraxgk.parallel) ParallelIdentityReport (class in spectraxgk.parallel) periodic_streaming_reference() (in module spectraxgk.velocity_sharding) periodic_streaming_shard_map() (in module spectraxgk.velocity_sharding) phase_align_eigenfunction() (in module spectraxgk.benchmarking)

phi_norm2() (in module spectraxgk.quasilinear) plot_saved_output() (in module spectraxgk.plotting) portfolio_objective_weight_vector() (in module spectraxgk.stellarator_objective_portfolio) portfolio_sample_weight_tensor() (in module spectraxgk.stellarator_objective_portfolio) production_independent_batching (spectraxgk.parallel_decomposition.DecompositionContract property) projected_line_search_input_manifest() (in module spectraxgk.vmec_jax_transport_line_search) ProjectedLineSearchPolicy (class in spectraxgk.vmec_jax_transport_line_search)

Q

qa_low_turbulence_comparison_payload() (in module spectraxgk.qa_low_turbulence) qa_low_turbulence_heat_flux_trace() (in module spectraxgk.qa_low_turbulence) qa_low_turbulence_objective() (in module spectraxgk.qa_low_turbulence) qa_low_turbulence_observable_sensitivity_report() (in module spectraxgk.qa_low_turbulence) qa_low_turbulence_observable_vector() (in module spectraxgk.qa_low_turbulence) qa_low_turbulence_observables() (in module spectraxgk.qa_low_turbulence) qa_low_turbulence_residual_names() (in module spectraxgk.qa_low_turbulence) qa_low_turbulence_residual_vector() (in module spectraxgk.qa_low_turbulence)

qa_low_turbulence_window_metrics() (in module spectraxgk.qa_low_turbulence) QALowTurbulenceConfig (class in spectraxgk.qa_low_turbulence) QALowTurbulenceResult (class in spectraxgk.qa_low_turbulence) quasilinear_calibration_report() (in module spectraxgk.quasilinear_calibration) quasilinear_feature_objective() (in module spectraxgk.quasilinear) QuasilinearCalibrationPoint (class in spectraxgk.quasilinear_calibration) QuasilinearTransportResult (class in spectraxgk.quasilinear) quasineutrality_phi() (in module spectraxgk.linear_moments)

R

read_external_holdout_screen() (in module spectraxgk.external_holdout_plan) real_fft_mesh() (in module spectraxgk.grids) real_fft_ordered_kx() (in module spectraxgk.grids) real_fft_unique_ky() (in module spectraxgk.grids) reconstruct_serial() (in module spectraxgk.parallel_decomposition) ReconstructionIdentityReport (class in spectraxgk.parallel_decomposition) reduced_boundary_surface() (in module spectraxgk.qa_low_turbulence) reduced_lcfs_bmag() (in module spectraxgk.qa_low_turbulence) reduced_portfolio_artifact_guard_report() (in module spectraxgk.stellarator_objective_portfolio) ReducedPortfolioArtifactGuardConfig (class in spectraxgk.stellarator_objective_portfolio) ReferenceSeries (class in spectraxgk.plotting) resolve_cfl_fac() (in module spectraxgk.config) resolve_nonlinear_artifact_policy() (in module spectraxgk.runtime_orchestration) resolve_scan_mode_method() (in module spectraxgk.benchmark_scan) rho_star (spectraxgk.normalization.NormalizationContract attribute) row_shape (spectraxgk.stellarator_objective_portfolio.StellaratorObjectivePortfolioContract property) run_cyclone_linear() (in module spectraxgk.benchmarks) run_cyclone_scan() (in module spectraxgk.benchmarks) run_etg_linear() (in module spectraxgk.benchmarks) run_etg_scan() (in module spectraxgk.benchmarks) run_kbm_beta_scan() (in module spectraxgk.benchmarks) run_kbm_linear() (in module spectraxgk.benchmarks) run_kbm_scan() (in module spectraxgk.benchmarks) run_kinetic_linear() (in module spectraxgk.benchmarks) run_kinetic_scan() (in module spectraxgk.benchmarks) run_linear_case() (in module spectraxgk.runtime) run_linear_scan() (in module spectraxgk.benchmarking)

run_nonlinear_case() (in module spectraxgk.runtime) run_runtime_linear() (in module spectraxgk.runtime) run_runtime_nonlinear() (in module spectraxgk.runtime) run_runtime_nonlinear_artifact_handoff() (in module spectraxgk.runtime_orchestration) run_runtime_scan() (in module spectraxgk.runtime) run_runtime_scan_batch() (in module spectraxgk.runtime_orchestration) run_scan_and_mode() (in module spectraxgk.benchmarking) run_tem_linear() (in module spectraxgk.benchmarks) run_tem_scan() (in module spectraxgk.benchmarks) RuntimeArtifactHandoffDeps (class in spectraxgk.runtime_orchestration) RuntimeCollisionConfig (class in spectraxgk.runtime_config) RuntimeConfig (class in spectraxgk.runtime_config) RuntimeExpertConfig (class in spectraxgk.runtime_config) RuntimeIndependentParallelPlan (class in spectraxgk.runtime) (class in spectraxgk.runtime_policies) RuntimeLinearResult (class in spectraxgk.runtime) RuntimeLinearScanResult (class in spectraxgk.runtime) RuntimeNonlinearResult (class in spectraxgk.runtime) RuntimeNormalizationConfig (class in spectraxgk.runtime_config) RuntimeOutputConfig (class in spectraxgk.runtime_config) RuntimeParallelConfig (class in spectraxgk.runtime_config) RuntimePhysicsConfig (class in spectraxgk.runtime_config) RuntimeProgressSnapshot (class in spectraxgk.runtime_orchestration) RuntimeQuasilinearConfig (class in spectraxgk.runtime_config) RuntimeScanBatchDeps (class in spectraxgk.runtime_orchestration) RuntimeSpeciesConfig (class in spectraxgk.runtime_config) RuntimeTermsConfig (class in spectraxgk.runtime_config)

S

SAlphaGeometry (class in spectraxgk.geometry) sample_flux_tube_geometry() (in module spectraxgk.geometry) sample_shape (spectraxgk.stellarator_objective_portfolio.StellaratorObjectivePortfolioContract property) saturated_flux_from_linear_weight() (in module spectraxgk.quasilinear) saturation_amplitude2() (in module spectraxgk.quasilinear) save_eigenfunction_reference_bundle() (in module spectraxgk.benchmarking) ScalarGateResult (class in spectraxgk.validation_gates) scan_comparison_figure() (in module spectraxgk.plotting) scan_multi_reference_figure() (in module spectraxgk.plotting) scan_window_valid() (in module spectraxgk.benchmark_scan) ScanAndModeResult (class in spectraxgk.benchmarking) ScanFitWindowPolicy (class in spectraxgk.benchmark_scan) select_admitted_transport_candidate() (in module spectraxgk.vmec_jax_transport_admission) select_fit_window() (in module spectraxgk.analysis) select_fit_window_loglinear() (in module spectraxgk.analysis) select_gx_real_fft_ky_grid() (in module spectraxgk.grids) select_kbm_solver_auto() (in module spectraxgk.benchmarks) select_ky_grid() (in module spectraxgk.grids) select_ky_index() (in module spectraxgk.analysis) select_projected_line_search_candidate() (in module spectraxgk.vmec_jax_transport_line_search) select_real_fft_ky_grid() (in module spectraxgk.grids) serial_reconstruction_identity_report() (in module spectraxgk.parallel_decomposition) set_plot_style() (in module spectraxgk.plotting) shape_aware_power_law_objective() (in module spectraxgk.quasilinear) shard_sequence() (in module spectraxgk.parallel_decomposition) ShardAssignment (class in spectraxgk.parallel_decomposition) shift_axis() (in module spectraxgk.linear_moments) should_use_ky_batch() (in module spectraxgk.benchmark_scan) size (spectraxgk.parallel_decomposition.ShardAssignment property) SlabGeometry (class in spectraxgk.geometry) sparse_descent_direction_from_gradient_report() (in module spectraxgk.vmec_jax_transport_line_search) spectral_phi_weights() (in module spectraxgk.quasilinear) SpectralGrid (class in spectraxgk.grids) spectrax_transport_objective_tuple() (in module spectraxgk.vmec_jax_transport_objective) spectraxgk.analysis module spectraxgk.autodiff_validation module spectraxgk.basis module spectraxgk.benchmark_defaults module spectraxgk.benchmark_scan module spectraxgk.benchmarking module spectraxgk.benchmarks module spectraxgk.config module spectraxgk.external_holdout_plan module spectraxgk.geometry module spectraxgk.geometry.differentiable module spectraxgk.grids module spectraxgk.gyroaverage module spectraxgk.linear module spectraxgk.linear_cache module spectraxgk.linear_linked module spectraxgk.linear_moments module

spectraxgk.linear_parallel module spectraxgk.linear_params module spectraxgk.nonlinear_diagnostics module spectraxgk.nonlinear_gradient_followup module spectraxgk.nonlinear_helpers module spectraxgk.nonlinear_replicate_diagnostics module spectraxgk.nonlinear_replicate_followup module spectraxgk.normalization module spectraxgk.operators module spectraxgk.parallel module spectraxgk.parallel_decomposition module spectraxgk.plotting module spectraxgk.qa_low_turbulence module spectraxgk.quasilinear module spectraxgk.quasilinear_calibration module spectraxgk.quasilinear_model_selection module spectraxgk.runtime module spectraxgk.runtime_config module spectraxgk.runtime_orchestration module spectraxgk.runtime_policies module spectraxgk.sharded_integrators module spectraxgk.stellarator_objective_portfolio module spectraxgk.validation_gates module spectraxgk.velocity_sharding module spectraxgk.vmec_jax_boundary_chain module spectraxgk.vmec_jax_candidate_gate module spectraxgk.vmec_jax_transport_admission module spectraxgk.vmec_jax_transport_gradient module spectraxgk.vmec_jax_transport_line_search module spectraxgk.vmec_jax_transport_objective module spectraxgk.zonal_objective module split_evenly() (in module spectraxgk.parallel) StellaratorObjectivePortfolioContract (class in spectraxgk.stellarator_objective_portfolio) streaming_term() (in module spectraxgk.linear_moments) sum_Jl2() (in module spectraxgk.gyroaverage)

T

TEMBaseCase (class in spectraxgk.config) TEMModelConfig (class in spectraxgk.config) term_config_to_linear_terms() (in module spectraxgk.linear_params) TimeConfig (class in spectraxgk.config) to_dict() (spectraxgk.external_holdout_plan.ExternalHoldoutScreenRow method) (spectraxgk.parallel.IndependentEnsembleProvenanceReport method) (spectraxgk.parallel.IndependentWorkerMetadata method) (spectraxgk.parallel.ParallelIdentityReport method) (spectraxgk.parallel_decomposition.DecompositionContract method) (spectraxgk.parallel_decomposition.ReconstructionIdentityReport method) (spectraxgk.parallel_decomposition.ShardAssignment method) (spectraxgk.qa_low_turbulence.QALowTurbulenceResult method) (spectraxgk.quasilinear.QuasilinearTransportResult method) (spectraxgk.runtime.RuntimeIndependentParallelPlan method) (spectraxgk.runtime_policies.RuntimeIndependentParallelPlan method) (spectraxgk.stellarator_objective_portfolio.ReducedPortfolioArtifactGuardConfig method) (spectraxgk.stellarator_objective_portfolio.StellaratorObjectivePortfolioContract method) (spectraxgk.vmec_jax_transport_admission.VMECJAXNonlinearAuditPolicy method) (spectraxgk.vmec_jax_transport_admission.VMECJAXNonlinearCampaignPolicy method) (spectraxgk.vmec_jax_transport_admission.VMECJAXReducedPrelaunchPolicy method) (spectraxgk.vmec_jax_transport_admission.VMECJAXTransportAdmissionPolicy method) (spectraxgk.vmec_jax_transport_line_search.ProjectedLineSearchPolicy method) (spectraxgk.zonal_objective.ZonalFlowObjectiveConfig method)

to_objective_term() (spectraxgk.vmec_jax_transport_objective.VMECJAXSpectraxTransportObjective method) transport_objective_sample_summary() (in module spectraxgk.vmec_jax_transport_admission) trim_terminal_theta_point() (spectraxgk.geometry.FluxTubeGeometryData method) twist_shift_params() (in module spectraxgk.geometry) twothirds_mask() (in module spectraxgk.grids)

U

unstable (spectraxgk.external_holdout_plan.ExternalHoldoutScreenRow property)

V

validate_objective_portfolio_contract() (in module spectraxgk.stellarator_objective_portfolio) velocity_field_reduce_reference() (in module spectraxgk.velocity_sharding) velocity_field_reduce_shard_map() (in module spectraxgk.velocity_sharding) VelocityShardingPlan (class in spectraxgk.velocity_sharding) vmec_boundary_aspect_sensitivity_report() (in module spectraxgk.geometry.differentiable) vmec_field_line_tensor_observable_names() (in module spectraxgk.geometry.differentiable) vmec_jax_boozer_equal_arc_core_profiles_from_state() (in module spectraxgk.geometry.differentiable) vmec_jax_boozer_flux_tube_sensitivity_report() (in module spectraxgk.geometry.differentiable) vmec_jax_field_line_tensor_sensitivity_report() (in module spectraxgk.geometry.differentiable) vmec_jax_flux_tube_array_parity_report() (in module spectraxgk.geometry.differentiable) vmec_jax_flux_tube_mapping_from_state() (in module spectraxgk.geometry.differentiable)

vmec_jax_flux_tube_sensitivity_report() (in module spectraxgk.geometry.differentiable) vmec_jax_metric_tensor_sensitivity_report() (in module spectraxgk.geometry.differentiable) vmec_jax_transport_growth_branch_locality_report_from_states() (in module spectraxgk.vmec_jax_transport_objective) vmec_jax_transport_objective_from_state() (in module spectraxgk.vmec_jax_transport_objective) vmec_metric_tensor_observable_names() (in module spectraxgk.geometry.differentiable) VMECJAXNonlinearAuditPolicy (class in spectraxgk.vmec_jax_transport_admission) VMECJAXNonlinearCampaignPolicy (class in spectraxgk.vmec_jax_transport_admission) VMECJAXReducedPrelaunchPolicy (class in spectraxgk.vmec_jax_transport_admission) VMECJAXSpectraxTransportObjective (class in spectraxgk.vmec_jax_transport_objective) VMECJAXTransportAdmissionPolicy (class in spectraxgk.vmec_jax_transport_admission) VMECJAXTransportObjectiveConfig (class in spectraxgk.vmec_jax_transport_objective)

W

windowed_nonlinear_metrics() (in module spectraxgk.benchmarking)

write_boundary_transport_gradient_report() (in module spectraxgk.vmec_jax_transport_gradient) write_quasilinear_calibration_report() (in module spectraxgk.quasilinear_calibration)

Z

zero_shear_enabled() (in module spectraxgk.geometry) zonal_flow_objective_artifact_from_records() (in module spectraxgk.zonal_objective) zonal_flow_objective_rows() (in module spectraxgk.zonal_objective) zonal_flow_objective_sensitivity_report() (in module spectraxgk.zonal_objective) zonal_flow_reduced_objective() (in module spectraxgk.zonal_objective)

zonal_flow_response_figure() (in module spectraxgk.plotting) zonal_flow_response_metrics() (in module spectraxgk.benchmarking) zonal_response_gate_report() (in module spectraxgk.validation_gates) ZonalFlowObjectiveConfig (class in spectraxgk.zonal_objective) ZonalFlowResponseMetrics (class in spectraxgk.validation_gates)