Index

_

_append_device_z_transport_observables() (in module gkx.operators.nonlinear.device_z) _append_spectral_physical_observable_vector() (in module gkx.operators.nonlinear.device_z) _apply_collision_split() (in module gkx.operators.nonlinear.collisions) _apply_completed_step_state_mask() (in module gkx.solvers.time.explicit) (in module gkx.solvers.time.explicit_steps) _apply_runtime_section_overrides() (in module gkx.workflows.runtime.toml) _arg_or_section() (in module gkx.workflows.runtime.commands) _as_runtime_species_array() (in module gkx.workflows.runtime.initial_phi) (in module gkx.workflows.runtime.startup) _asdict() (gkx.operators.linear.dissipation.CollisionInvariantRates method) (gkx.operators.nonlinear.projection.ShearingCoordinateUpdate method) _bgrad_from_bmag() (in module gkx.geometry.core) (in module gkx.geometry.flux_tube) _blocked_device_z_rhs_report() (in module gkx.operators.nonlinear.device_z) _blocked_device_z_transport_window_report() (in module gkx.operators.nonlinear.device_z) _boozer_half_mesh_s_grid() (in module gkx.geometry.numerics) _build_initial_condition() (in module gkx.workflows.runtime.startup) _build_linked_end_damping_profile() (in module gkx.operators.linear.linked) _build_linked_fft_maps() (in module gkx.operators.linear.linked) _build_output_grid_and_geometry() (in module gkx.artifacts.nonlinear_netcdf) _build_shift_invert_precond() (in module gkx.solvers.linear.krylov) _build_single_phi_gaussian_profile() (in module gkx.workflows.runtime.initial_conditions) (in module gkx.workflows.runtime.startup) _cached_booz_xform_constants() (in module gkx.geometry.vmec_boozer_constants) _cached_hermitian_mode_weight() (in module gkx.diagnostics.moments) _candidate_paths() (in module gkx.geometry.backend_discovery) _centered_glibc_random_pairs() (in module gkx.workflows.runtime.initial_conditions) (in module gkx.workflows.runtime.startup) _chunk_end_time() (in module gkx.workflows.runtime.chunks) _collision_damping() (in module gkx.operators.nonlinear.collisions) _completed_step_state_mask() (in module gkx.solvers.time.explicit) (in module gkx.solvers.time.explicit_steps) _condense_kx_for_output() (in module gkx.artifacts.spectral_layout) _condense_ky_for_output() (in module gkx.artifacts.spectral_layout) _condense_kykx_for_output() (in module gkx.artifacts.spectral_layout) _conditioning_gate_from_covariance() (in module gkx.objectives.stellarator_reduced) _dealiased_initial_mode_pairs() (in module gkx.workflows.runtime.initial_conditions) (in module gkx.workflows.runtime.startup) _density_moments_for_target_phi() (in module gkx.workflows.runtime.initial_phi) (in module gkx.workflows.runtime.startup) _device_z_pencil_shard_map_observables_fn() (in module gkx.operators.nonlinear.device_z) _device_z_pencil_shard_map_rhs_fn() (in module gkx.operators.nonlinear.device_z) _device_z_rhs_identity_report() (in module gkx.operators.nonlinear.device_z) _device_z_sharding_for_spectral_state() (in module gkx.operators.nonlinear.device_z) _device_z_transport_identity_passed() (in module gkx.operators.nonlinear.device_z) _device_z_transport_window_report() (in module gkx.operators.nonlinear.device_z) _effective_diagnostics_stride() (in module gkx.workflows.runtime.chunks) _electrostatic_streaming_field_rhs() (in module gkx.solvers.linear.parallel) (in module gkx.solvers.linear.parallel_streaming) _enforce_full_ky_hermitian() (in module gkx.workflows.runtime.initial_conditions) (in module gkx.workflows.runtime.startup) _fft2_xy_remove_radial_phase() (in module gkx.operators.nonlinear.brackets) _find_importable_module() (in module gkx.geometry.backend_discovery) _growth_rate_mode_mask() (in module gkx.solvers.time.explicit) (in module gkx.solvers.time.explicit_steps) _hermitian_mode_weight() (in module gkx.diagnostics.moments) _host_max_abs_rel_error() (in module gkx.operators.nonlinear.spectral_core) _host_staged_array_for_sharding() (in module gkx.operators.nonlinear.spectral_core) _ifft2_xy_with_radial_phase() (in module gkx.operators.nonlinear.brackets) _ImportedGeometrySelection (class in gkx.geometry.flux_tube) _infer_root_theta_closed_interval() (in module gkx.geometry.flux_tube) _infer_runtime_nonlinear_steps() (in module gkx.workflows.runtime.policies) _InitialConditionBuilder (class in gkx.workflows.runtime.initial_conditions) _instantaneous_growth_rate_step() (in module gkx.solvers.time.explicit) (in module gkx.solvers.time.explicit_steps) _integrate_linear_cached_impl() (in module gkx.solvers.linear.integrators) _integrate_linear_implicit_cached() (in module gkx.solvers.linear.implicit) _interp_equal_arc_profile() (in module gkx.geometry.numerics) _is_traced() (in module gkx.geometry.backend_discovery) _jax_float_dtype() (in module gkx.geometry.backend_discovery) _jl_family() (in module gkx.diagnostics.moments) _json_ready() (in module gkx.geometry.autodiff_checks) _LaguerreGridContext (class in gkx.terms.nonlinear) _linear_explicit_step() (in module gkx.solvers.time.explicit) (in module gkx.solvers.time.explicit_steps) _linear_rhs_electrostatic_slices_velocity_sharded_fused() (in module gkx.solvers.linear.parallel) (in module gkx.solvers.linear.parallel_electrostatic) _load_vmex_example_state_bundle() (in module gkx.objectives.vmec_boozer_fd) _log_amp_phase() (in module gkx.diagnostics.growth_rates) _make() (gkx.operators.linear.dissipation.CollisionInvariantRates class method) (gkx.operators.nonlinear.projection.ShearingCoordinateUpdate class method) _make_fixed_mode_projector() (in module gkx.operators.nonlinear.projection)

_make_hermitian_projector() (in module gkx.operators.nonlinear.projection) _make_nonlinear_state_projector() (in module gkx.operators.nonlinear.projection) _masked_abs2() (in module gkx.diagnostics.moments) _merge_dataclass() (in module gkx.workflows.runtime.toml) _mode21_vmec_boozer_linear_context() (in module gkx.objectives.vmec_boozer_gradients) _mode21_vmec_boozer_quasilinear_features() (in module gkx.objectives.vmec_boozer_gradients) _mode_family_sign() (in module gkx.solvers.linear.krylov) _new_transport_trace_dict() (in module gkx.operators.nonlinear.device_z) _next_elapsed_time() (in module gkx.workflows.runtime.chunks) _NonlinearBracketContext (class in gkx.terms.nonlinear) _normalize_geometry_overrides() (in module gkx.workflows.runtime.toml) _normalize_spectral_tile_bounds() (in module gkx.operators.nonlinear.spectral_core) _offset_chunk_diagnostics_time() (in module gkx.workflows.runtime.chunks) _omega_scale() (in module gkx.solvers.linear.krylov) _parallel_requests_combined_ky_scan() (in module gkx.workflows.runtime.policies) _parse_ky_values() (in module gkx.workflows.runtime.commands) _pencil_fft2() (in module gkx.operators.nonlinear.spectral_core) _pencil_ifft2() (in module gkx.operators.nonlinear.spectral_core) _pencil_spectral_bracket() (in module gkx.operators.nonlinear.spectral_core) _pencil_spectral_bracket_z_chunked() (in module gkx.operators.nonlinear.spectral_core) _periodic_bilinear_sample_2d() (in module gkx.geometry.numerics) _periodic_spectral_derivative() (in module gkx.geometry.core) (in module gkx.geometry.flux_tube) _periodic_zp_from_grid() (in module gkx.workflows.runtime.initial_conditions) (in module gkx.workflows.runtime.startup) _PhiMomentAlgebra (class in gkx.workflows.runtime.initial_phi) _physical_omega() (in module gkx.solvers.linear.krylov) _precision_gate_tolerances() (in module gkx.objectives.stellarator_reduced) _prepare_nonlinear_path() (in module gkx.terms.nonlinear) _prepare_runtime_command_config() (in module gkx.workflows.runtime.commands) _PreparedNonlinearInputs (class in gkx.terms.nonlinear) _PreparedNonlinearPath (class in gkx.terms.nonlinear) _qa_core_features() (in module gkx.objectives.stellarator_reduced) _qa_geometry_features() (in module gkx.objectives.stellarator_reduced) _qa_gradient_drives() (in module gkx.objectives.stellarator_reduced) _qa_linear_itg_features() (in module gkx.objectives.stellarator_reduced) _qa_quasilinear_heat_flux() (in module gkx.objectives.stellarator_reduced) _reduce_scalar_kykxz() (in module gkx.diagnostics.moments) _reduce_species_kykxz() (in module gkx.diagnostics.moments) _replace() (gkx.operators.linear.dissipation.CollisionInvariantRates method) (gkx.operators.nonlinear.projection.ShearingCoordinateUpdate method) _replace_runtime_section() (in module gkx.workflows.runtime.toml) _report_float() (in module gkx.objectives.vmec_boozer_fd) _residual_precision_gate_tolerances() (in module gkx.objectives.stellarator_reduced) _resolve_grid_time_options() (in module gkx.workflows.runtime.commands) _resolve_linear_command_options() (in module gkx.workflows.runtime.commands) _resolve_linear_fit_options() (in module gkx.workflows.runtime.commands) _resolve_nonlinear_command_options() (in module gkx.workflows.runtime.commands) _resolve_parallel_devices() (in module gkx.solvers.linear.parallel) (in module gkx.solvers.linear.parallel_common) _resolve_runtime_config_paths() (in module gkx.workflows.runtime.toml) _resolve_runtime_vmec_file() (in module gkx.geometry.vmec_eik) _resolve_scan_command_options() (in module gkx.workflows.runtime.commands) _rk3_heun_step() (in module gkx.solvers.time.explicit) (in module gkx.solvers.time.explicit_steps) _rk4_step() (in module gkx.solvers.time.explicit) (in module gkx.solvers.time.explicit_steps) _run_device_z_compute_window_states() (in module gkx.operators.nonlinear.device_z) _runtime_base_config() (in module gkx.workflows.runtime.toml) _runtime_external_phi() (in module gkx.workflows.runtime.policies) _runtime_fit_config() (in module gkx.workflows.runtime.commands) _runtime_independent_parallel_plan() (in module gkx.workflows.runtime.policies) _runtime_species_from_toml() (in module gkx.workflows.runtime.toml) _RuntimeLinearRequest (class in gkx.workflows.runtime.execution) _sampled_qa_itg_fields() (in module gkx.objectives.stellarator_reduced) _select_by_overlap() (in module gkx.solvers.linear.krylov) _select_by_target() (in module gkx.solvers.linear.krylov) _select_imported_geometry_variables() (in module gkx.geometry.flux_tube) _sensitivity_conditioning_metadata() (in module gkx.geometry.autodiff_checks) _serial_transport_compute_state() (in module gkx.operators.nonlinear.device_z) _spectral_layout_round_trip() (in module gkx.operators.nonlinear.spectral_core) _spectral_physical_transport_observable_sums() (in module gkx.operators.nonlinear.device_z) _spectral_physical_transport_observable_vector_from_sums() (in module gkx.operators.nonlinear.device_z) _spectral_rhs_from_bracket() (in module gkx.operators.nonlinear.spectral_core) _split_vmec_boozer_objective_kwargs() (in module gkx.objectives.vmec_boozer) _streaming_electrostatic_from_phi_velocity_sharded() (in module gkx.solvers.linear.parallel) (in module gkx.solvers.linear.parallel_streaming) _transport_mode_weight() (in module gkx.diagnostics.moments) _transport_trace_error_pairs() (in module gkx.operators.nonlinear.device_z) _transport_trace_tuples() (in module gkx.operators.nonlinear.device_z) _validate_nonlinear_config_state_sharding() (in module gkx.solvers.time.runners) _within_abs_or_rel_tolerance() (in module gkx.operators.nonlinear.spectral_core) _write_diagnostics_group() (in module gkx.artifacts.nonlinear_netcdf)

A

AdaptiveChunkResult (class in gkx.workflows.runtime.chunks) advance_explicit_nonlinear_state() (in module gkx.solvers.nonlinear.explicit) advance_imex_nonlinear_state() (in module gkx.solvers.nonlinear.imex) advance_shearing_coordinates() (in module gkx.operators.nonlinear.projection) aggregate_objective_portfolio() (in module gkx.objectives.portfolio) apply() (gkx.operators.collision.CollisionOperator method) apply_diagnostic_normalization() (in module gkx.diagnostics.normalization) apply_geometry_grid_defaults() (in module gkx.geometry.core) apply_heat_flux_scale() (in module gkx.diagnostics.quasilinear_calibration)

apply_hermite_v() (in module gkx.operators.linear.moments) apply_hermite_v2() (in module gkx.operators.linear.moments) apply_imported_geometry_grid_defaults() (in module gkx.geometry.core) apply_laguerre_x() (in module gkx.operators.linear.moments) apply_quasilinear_overrides() (in module gkx.workflows.runtime.commands) apply_runtime_path_overrides() (in module gkx.workflows.runtime.commands) associated_bessel_laguerre_coefficients() (in module gkx.core.velocity) attach_preloaded_runtime_config() (in module gkx.workflows.runtime.commands) autodiff_finite_difference_report() (in module gkx.objectives.autodiff_validation)

B

batch_map() (in module gkx.parallel) (in module gkx.parallel.batch) batch_map_identity_report() (in module gkx.parallel) (in module gkx.parallel.batch) bessel_j0() (in module gkx.core.velocity) bessel_j1() (in module gkx.core.velocity) bessel_laguerre_kernels() (in module gkx.core.velocity) bgrad() (gkx.geometry.analytic.SAlphaGeometry method) (gkx.geometry.core.SAlphaGeometry method) bmag() (gkx.geometry.analytic.SAlphaGeometry method) (gkx.geometry.core.SAlphaGeometry method) booz_xform_flux_tube_mapping_from_inputs() (in module gkx.geometry.booz_xform_bridge) (in module gkx.geometry.differentiable) booz_xform_flux_tube_sensitivity_report() (in module gkx.geometry.booz_xform_bridge) (in module gkx.geometry.differentiable) booz_xform_spectral_sensitivity_report() (in module gkx.geometry.booz_xform_bridge) (in module gkx.geometry.differentiable) boundary_chain_accepted_parameter_indices() (in module gkx.objectives.vmec_transport_optimization) boundary_chain_summary_from_probe() (in module gkx.geometry.vmec_boundary_chain) boundary_spec_record() (in module gkx.objectives.vmec_transport_optimization) branch_continuity_gate_report() (in module gkx.diagnostics.validation_gates) build_authoritative_wout_candidate_gate() (in module gkx.objectives.vmec_candidate_admission) build_boundary_chain_collection_summary() (in module gkx.geometry.vmec_boundary_chain) build_boundary_chain_summary() (in module gkx.geometry.vmec_boundary_chain) build_boundary_transport_gradient_report() (in module gkx.objectives.vmec_transport_optimization) build_collocation_surfaces() (in module gkx.geometry.analytic) build_diagnostic_nonlinear_domain_decomposition() (in module gkx.parallel) (in module gkx.parallel.decomposition) build_flux_tube_geometry() (in module gkx.geometry.core) build_H() (in module gkx.operators.linear.moments) build_independent_portfolio_decomposition() (in module gkx.parallel) (in module gkx.parallel.decomposition)

build_linear_params() (in module gkx.operators.linear.params) build_miller_geometry_request() (in module gkx.geometry.miller_eik) build_nonlinear_audit_redesign_report() (in module gkx.diagnostics.stellarator_transport_reports) build_nonlinear_campaign_admission_report() (in module gkx.diagnostics.stellarator_transport_reports) build_nonlinear_collision_split_policy() (in module gkx.operators.nonlinear.collisions) build_nonlinear_domain_decomposition_plan() (in module gkx.operators.nonlinear.domain_decomposition) build_nonlinear_landscape_admission_report() (in module gkx.diagnostics.stellarator_transport_reports) build_quasilinear_model_selection_status() (in module gkx.diagnostics.quasilinear_model_selection) build_quasilinear_model_selection_status_from_paths() (in module gkx.diagnostics.quasilinear_model_selection) build_reduced_nonlinear_audit_prelaunch_report() (in module gkx.diagnostics.stellarator_transport_reports) build_runtime_command_deps() (in module gkx.workflows.runtime.commands) build_runtime_geometry() (in module gkx.runtime) (in module gkx.workflows.runtime.startup) build_runtime_linear_dispatch_deps() (in module gkx.workflows.runtime.execution) build_runtime_linear_params() (in module gkx.runtime) (in module gkx.workflows.runtime.startup) build_runtime_linear_terms() (in module gkx.runtime) (in module gkx.workflows.runtime.startup) build_runtime_nonlinear_diagnostics_kwargs() (in module gkx.workflows.runtime.policies) build_runtime_nonlinear_dispatch_deps() (in module gkx.workflows.runtime.execution) build_runtime_nonlinear_result() (in module gkx.workflows.runtime.results) build_runtime_progress_message() (in module gkx.workflows.runtime.chunks) build_runtime_scan_batch_deps() (in module gkx.workflows.runtime.orchestration_scan) build_runtime_scan_orchestration_deps() (in module gkx.workflows.runtime.orchestration_scan) build_runtime_term_config() (in module gkx.runtime) (in module gkx.workflows.runtime.startup) build_solved_vmec_candidate_gate() (in module gkx.objectives.vmec_candidate_admission) build_transport_admission_report() (in module gkx.objectives.vmec_transport_admission) build_velocity_sharding_plan() (in module gkx.parallel) (in module gkx.parallel.velocity_plan) build_vmec_geometry_request() (in module gkx.geometry.vmec_eik) build_wout_reproducibility_gate() (in module gkx.objectives.vmec_candidate_admission)

C

cache (gkx.operators.collision.CollisionContext attribute) calibration_point_from_nonlinear_window_summary() (in module gkx.diagnostics.quasilinear_calibration) calibration_point_from_spectrum_and_nonlinear_window() (in module gkx.diagnostics.quasilinear_calibration) candidate_transport_metric() (in module gkx.objectives.vmec_transport_admission) case (gkx.diagnostics.normalization.NormalizationContract attribute) central_finite_difference_jacobian() (in module gkx.objectives.autodiff_validation) checkpoint_explicit_step() (in module gkx.solvers.nonlinear.explicit) classify_gradient_artifact() (in module gkx.diagnostics.nonlinear_gradient_evidence) collision_invariant_rates() (in module gkx.operators.linear.dissipation) collision_quadratic_rate() (in module gkx.operators.linear.dissipation) CollisionContext (class in gkx.operators.collision) CollisionInvariantRates (class in gkx.operators.linear.dissipation)

CollisionOperator (class in gkx.operators.collision) compare_eigenfunctions() (in module gkx.diagnostics.modes) compute_b() (in module gkx.operators.linear.moments) compute_nonlinear_diagnostic_tuple() (in module gkx.operators.nonlinear.diagnostic_state) compute_quasilinear_from_linear_state() (in module gkx.diagnostics.quasilinear_transport) concat_runtime_diagnostics() (in module gkx.workflows.runtime.diagnostic_arrays) covariance_diagnostics() (in module gkx.objectives.autodiff_validation) cumulative_mode_shift (gkx.operators.nonlinear.projection.ShearingCoordinateUpdate attribute) curvature_gradb_drift_reference() (in module gkx.parallel) (in module gkx.parallel.velocity_streaming) curvature_gradb_drift_shard_map() (in module gkx.parallel) (in module gkx.parallel.velocity_streaming) CycloneBaseCase (class in gkx.config)

D

DecompositionContract (class in gkx.parallel) (class in gkx.parallel.decomposition) default_miller_eik_output_path() (in module gkx.geometry.miller_eik) default_solver_geometry_design_params() (in module gkx.objectives.geometry) default_stellarator_initial_params() (in module gkx.objectives.stellarator_reduced) default_vmec_eik_output_path() (in module gkx.geometry.vmec_eik) density (gkx.operators.linear.dissipation.CollisionInvariantRates attribute) density_moment() (in module gkx.diagnostics.modes) deterministic_nonlinear_domain_state() (in module gkx.operators.nonlinear.domain_decomposition) deterministic_nonlinear_spectral_state() (in module gkx.operators.nonlinear.spectral_core) device_z_pencil_fft_batch_pressure_model() (in module gkx.operators.nonlinear.spectral_core) device_z_pencil_nonlinear_spectral_rhs() (in module gkx.operators.nonlinear.device_z) device_z_pencil_nonlinear_spectral_transport_window_identity_gate() (in module gkx.operators.nonlinear.device_z) diagnostic_nonlinear_partition (gkx.parallel.decomposition.DecompositionContract property) (gkx.parallel.DecompositionContract property) diagnostic_norm_default (gkx.diagnostics.normalization.NormalizationContract attribute) DiagnosticTimeSeries (class in gkx.artifacts.io) diamagnetic_drive_coeffs() (in module gkx.operators.linear.moments) diamagnetic_drive_reference() (in module gkx.parallel) (in module gkx.parallel.velocity_drive) diamagnetic_drive_shard_map() (in module gkx.parallel) (in module gkx.parallel.velocity_drive)

direct_config_shorthand_args() (in module gkx.workflows.runtime.toml) discover_differentiable_geometry_backends() (in module gkx.geometry.backend_discovery) (in module gkx.geometry.differentiable) distribution (gkx.operators.collision.CollisionContext attribute) distribution_free_energy() (in module gkx.diagnostics.moments) distribution_free_energy_resolved() (in module gkx.diagnostics.moments) dominant_eigenpair() (in module gkx.solvers.linear.krylov) dominant_eigenpair_cached() (in module gkx.solvers.linear.krylov) dominant_eigenpair_power() (in module gkx.solvers.linear.krylov) dominant_eigenpair_propagator_cached() (in module gkx.solvers.linear.krylov) dominant_eigenpair_shift_invert_cached() (in module gkx.solvers.linear.krylov) dominant_eigenvalue_branch_locality_report() (in module gkx.objectives.eigen) dominant_real_eigenvalue() (in module gkx.objectives.eigen) drift_coeffs() (gkx.geometry.analytic.SAlphaGeometry method) (gkx.geometry.core.SAlphaGeometry method) drift_components() (gkx.geometry.analytic.SAlphaGeometry method) (gkx.geometry.core.SAlphaGeometry method) drift_kinetic_coulomb_six_moment_contribution() (in module gkx.operators.linear.dissipation) drift_kinetic_dougherty_contribution() (in module gkx.operators.linear.dissipation) drift_kinetic_sugama_six_moment_contribution() (in module gkx.operators.linear.dissipation)

E

effective_boundary() (in module gkx.geometry.analytic) (in module gkx.geometry.core) effective_kperp2() (in module gkx.diagnostics.quasilinear_transport) effective_kx (gkx.operators.nonlinear.projection.ShearingCoordinateUpdate attribute) eigenfunction_gate_report() (in module gkx.diagnostics.validation_gates) EigenfunctionComparisonMetrics (class in gkx.diagnostics.modes) EigenfunctionReferenceBundle (class in gkx.diagnostics.modes) electrostatic_field_energy() (in module gkx.diagnostics.moments) electrostatic_field_energy_resolved() (in module gkx.diagnostics.moments) electrostatic_phi_reference() (in module gkx.parallel) (in module gkx.parallel.velocity_drive) electrostatic_phi_shard_map() (in module gkx.parallel) (in module gkx.parallel.velocity_drive) enabled (gkx.runtime.RuntimeIndependentParallelPlan property) (gkx.workflows.runtime.policies.RuntimeIndependentParallelPlan property)

energy_operator() (in module gkx.operators.linear.moments) ensure_flux_tube_geometry_data() (in module gkx.geometry.core) evaluate_boozer_bmag_on_field_line() (in module gkx.geometry.booz_xform_bridge) (in module gkx.geometry.differentiable) evaluate_scalar_gate() (in module gkx.diagnostics.validation_gates) exb_nonlinear_contribution() (in module gkx.terms.nonlinear) explicit_complex_operator_matrix() (in module gkx.objectives.autodiff_validation) explicit_method_default_cfl_fac() (in module gkx.config) ExplicitNonlinearDiagnosticsDeps (class in gkx.solvers.nonlinear.diagnostics) ExplicitTimeConfig (class in gkx.solvers.time.explicit) ExplicitTimeConfigLike (class in gkx.solvers.time.explicit_diagnostics) extract_eigenfunction() (in module gkx.diagnostics.modes) extract_mode() (in module gkx.diagnostics.modes) extract_mode_time_series() (in module gkx.diagnostics.modes)

F

fieldline_quadrature_weights() (in module gkx.diagnostics.moments) fields (gkx.operators.collision.CollisionContext attribute) final_iota_profiles_from_vmec_result() (in module gkx.objectives.vmec_candidate_admission) finalize_runtime_linear_quasilinear() (in module gkx.workflows.runtime.diagnostics) finite_difference_jacobian() (in module gkx.geometry.autodiff_checks) (in module gkx.geometry.differentiable) fit_growth_rate() (in module gkx.diagnostics.growth_rates) fit_growth_rate_auto() (in module gkx.diagnostics.growth_rates) fit_growth_rate_auto_with_stats() (in module gkx.diagnostics.growth_rates) fit_growth_rate_with_stats() (in module gkx.diagnostics.growth_rates)

fit_runtime_linear_diagnostics() (in module gkx.workflows.runtime.diagnostics) fit_train_heat_flux_scale() (in module gkx.diagnostics.quasilinear_calibration) flux_tube_geometry_from_mapping() (in module gkx.geometry.differentiable) (in module gkx.geometry.flux_tube_contract) flux_tube_geometry_from_vmec_boozer_state() (in module gkx.geometry.differentiable) (in module gkx.geometry.vmec_boozer_core) flux_tube_geometry_observables() (in module gkx.geometry.differentiable) (in module gkx.geometry.flux_tube_contract) FluxTubeGeometryData (class in gkx.geometry.core) (class in gkx.geometry.flux_tube) format_duration() (in module gkx.workflows.runtime.chunks)

G

gamma0() (in module gkx.core.velocity) gate_report() (in module gkx.diagnostics.validation_gates) gate_report_to_dict() (in module gkx.diagnostics.validation_gates) GateReport (class in gkx.diagnostics.validation_gates) generate_runtime_miller_eik() (in module gkx.geometry.miller_eik) generate_runtime_vmec_eik() (in module gkx.geometry.vmec_eik) geometry_inverse_design_report() (in module gkx.geometry.differentiable) (in module gkx.geometry.sensitivity) geometry_observable_names() (in module gkx.geometry.differentiable) (in module gkx.geometry.flux_tube_contract) geometry_sensitivity_report() (in module gkx.geometry.differentiable) (in module gkx.geometry.sensitivity) GeometryConfig (class in gkx.config) (class in gkx.geometry.core) get_normalization_contract() (in module gkx.diagnostics.normalization) gkx.artifacts module gkx.artifacts.io module gkx.artifacts.nonlinear_netcdf module gkx.artifacts.spectral_layout module gkx.artifacts.zonal_plots module gkx.config module gkx.core.grid module gkx.core.velocity module gkx.diagnostics.growth_rates module gkx.diagnostics.growth_windows module gkx.diagnostics.modes module gkx.diagnostics.moments module gkx.diagnostics.nonlinear_gradient_evidence module gkx.diagnostics.nonlinear_gradient_statistics module gkx.diagnostics.nonlinear_replicates module gkx.diagnostics.nonlinear_transport_optimization module gkx.diagnostics.normalization module gkx.diagnostics.quasilinear_calibration module gkx.diagnostics.quasilinear_model_selection module gkx.diagnostics.quasilinear_transport module gkx.diagnostics.stellarator_transport_reports module gkx.diagnostics.transport module gkx.diagnostics.transport_windows module gkx.diagnostics.validation_gates module gkx.geometry.analytic module gkx.geometry.autodiff_checks module gkx.geometry.backend_discovery module gkx.geometry.booz_xform_bridge module gkx.geometry.core module gkx.geometry.differentiable module gkx.geometry.flux_tube module gkx.geometry.flux_tube_contract module gkx.geometry.miller_eik module gkx.geometry.numerics module gkx.geometry.sensitivity module gkx.geometry.vmec_boozer_constants module gkx.geometry.vmec_boozer_core module gkx.geometry.vmec_boundary_chain module gkx.geometry.vmec_eik module gkx.geometry.vmec_flux_tube_reports module gkx.geometry.vmec_state_sensitivity module gkx.geometry.vmec_tensor_mapping module gkx.objectives.autodiff_validation module gkx.objectives.core module gkx.objectives.eigen module gkx.objectives.geometry module gkx.objectives.gradient_gates module gkx.objectives.portfolio module gkx.objectives.portfolio_guard module gkx.objectives.sampling module gkx.objectives.stellarator_contracts module gkx.objectives.stellarator_reduced module gkx.objectives.stellarator_tables module gkx.objectives.vmec_boozer module gkx.objectives.vmec_boozer_fd module gkx.objectives.vmec_boozer_gradients module gkx.objectives.vmec_boozer_line_search module gkx.objectives.vmec_candidate_admission module gkx.objectives.vmec_transport module gkx.objectives.vmec_transport_admission module gkx.objectives.vmec_transport_branch module

gkx.objectives.vmec_transport_optimization module gkx.objectives.zonal module gkx.operators module gkx.operators.collision module gkx.operators.linear module gkx.operators.linear.dissipation module gkx.operators.linear.linked module gkx.operators.linear.moments module gkx.operators.linear.params module gkx.operators.linear.rhs module gkx.operators.nonlinear.brackets module gkx.operators.nonlinear.collisions module gkx.operators.nonlinear.device_z module gkx.operators.nonlinear.diagnostic_state module gkx.operators.nonlinear.domain_decomposition module gkx.operators.nonlinear.projection module gkx.operators.nonlinear.rhs module gkx.operators.nonlinear.spectral_core module gkx.parallel module gkx.parallel.batch module gkx.parallel.decomposition module gkx.parallel.identity module gkx.parallel.independent module gkx.parallel.integrators module gkx.parallel.state module gkx.parallel.velocity_drive module gkx.parallel.velocity_hermite module gkx.parallel.velocity_plan module gkx.parallel.velocity_streaming module gkx.runtime module gkx.solvers.linear.implicit module gkx.solvers.linear.integrator_diagnostics module gkx.solvers.linear.integrators module gkx.solvers.linear.krylov module gkx.solvers.linear.parallel module gkx.solvers.linear.parallel_common module gkx.solvers.linear.parallel_electrostatic module gkx.solvers.linear.parallel_streaming module gkx.solvers.nonlinear.diagnostics module gkx.solvers.nonlinear.explicit module gkx.solvers.nonlinear.imex module gkx.solvers.time.diffrax_core module gkx.solvers.time.diffrax_linear module gkx.solvers.time.diffrax_nonlinear module gkx.solvers.time.diffrax_streaming module gkx.solvers.time.explicit module gkx.solvers.time.explicit_cfl module gkx.solvers.time.explicit_diagnostics module gkx.solvers.time.explicit_steps module gkx.solvers.time.runners module gkx.terms.nonlinear module gkx.workflows.runtime.chunks module gkx.workflows.runtime.commands module gkx.workflows.runtime.config module gkx.workflows.runtime.diagnostic_arrays module gkx.workflows.runtime.diagnostics module gkx.workflows.runtime.execution module gkx.workflows.runtime.initial_conditions module gkx.workflows.runtime.initial_phi module gkx.workflows.runtime.orchestration_artifacts module gkx.workflows.runtime.orchestration_scan module gkx.workflows.runtime.policies module gkx.workflows.runtime.results module gkx.workflows.runtime.startup module gkx.workflows.runtime.toml module grad_z_periodic() (in module gkx.operators.linear.moments) gradient_scope (gkx.objectives.vmec_transport.VMEXTransportObjectiveConfig property) gradpar() (gkx.geometry.analytic.SAlphaGeometry method) (gkx.geometry.core.SAlphaGeometry method) GridConfig (class in gkx.config) (class in gkx.geometry.core)

H

hamiltonian (gkx.operators.collision.CollisionContext attribute) heat_flux_channel_resolved_species() (in module gkx.diagnostics.moments) heat_flux_channel_species() (in module gkx.diagnostics.transport) heat_flux_resolved_species() (in module gkx.diagnostics.moments) heat_flux_species() (in module gkx.diagnostics.transport) heat_flux_total() (in module gkx.diagnostics.transport) hermite_ladder_coeffs() (in module gkx.core.velocity) hermite_neighbor_reference() (in module gkx.parallel) (in module gkx.parallel.velocity_hermite) hermite_neighbor_shard_map() (in module gkx.parallel) (in module gkx.parallel.velocity_hermite) hermite_normed() (in module gkx.core.velocity)

hermite_physicists() (in module gkx.core.velocity) hermite_shift_reference() (in module gkx.parallel) (in module gkx.parallel.velocity_hermite) hermite_shift_shard_map() (in module gkx.parallel) (in module gkx.parallel.velocity_hermite) hermite_streaming() (in module gkx.operators) (in module gkx.operators.linear.moments) hermite_streaming_ladder_reference() (in module gkx.parallel) (in module gkx.parallel.velocity_streaming) hermite_streaming_ladder_shard_map() (in module gkx.parallel) (in module gkx.parallel.velocity_streaming) hyperdiffusion_contribution() (in module gkx.operators.linear.dissipation)

I

imex_fixed_point_guess() (in module gkx.solvers.nonlinear.imex) IMEXNonlinearDiagnosticsDeps (class in gkx.solvers.nonlinear.diagnostics) implicit_eigenpair_observable_sensitivity_report() (in module gkx.objectives.autodiff_validation) incremental_mode_shift (gkx.operators.nonlinear.projection.ShearingCoordinateUpdate attribute) independent_ensemble_provenance_gate() (in module gkx.parallel) (in module gkx.parallel.independent) independent_map() (in module gkx.parallel) (in module gkx.parallel.independent) independent_map_identity_report() (in module gkx.parallel) (in module gkx.parallel.independent) independent_worker_metadata() (in module gkx.parallel) (in module gkx.parallel.independent) IndependentEnsembleProvenanceReport (class in gkx.parallel) (class in gkx.parallel.independent) IndependentMapExecutionError, [1] IndependentWorkerMetadata (class in gkx.parallel) (class in gkx.parallel.independent) infer_triple_dealiased_ny() (in module gkx.artifacts.spectral_layout) InitializationConfig (class in gkx.config) instantaneous_growth_rate_from_phi() (in module gkx.diagnostics.growth_rates) integrate_cached_explicit_scan() (in module gkx.solvers.nonlinear.explicit) integrate_cached_imex_scan() (in module gkx.solvers.nonlinear.imex) integrate_explicit_nonlinear_diagnostics_impl() (in module gkx.solvers.nonlinear.diagnostics)

integrate_imex_nonlinear_diagnostics_impl() (in module gkx.solvers.nonlinear.diagnostics) integrate_linear() (in module gkx.solvers.linear.integrators) integrate_linear_diagnostics() (in module gkx.solvers.linear.integrator_diagnostics) (in module gkx.solvers.linear.integrators) integrate_linear_diffrax() (in module gkx.solvers.time.diffrax_linear) integrate_linear_diffrax_streaming() (in module gkx.solvers.time.diffrax_streaming) integrate_linear_explicit() (in module gkx.solvers.time.explicit) integrate_linear_explicit_diagnostics() (in module gkx.solvers.time.explicit) (in module gkx.solvers.time.explicit_diagnostics) integrate_linear_explicit_from_config() (in module gkx.solvers.time.explicit) integrate_linear_from_config() (in module gkx.solvers.time.runners) integrate_linear_sharded() (in module gkx.parallel) (in module gkx.parallel.integrators) integrate_nonlinear_diffrax() (in module gkx.solvers.time.diffrax_nonlinear) integrate_nonlinear_from_config() (in module gkx.solvers.time.runners) integrate_nonlinear_scan() (in module gkx.solvers.nonlinear.explicit) integrate_nonlinear_sharded() (in module gkx.parallel) (in module gkx.parallel.integrators) integrated_quasilinear_flux_from_spectrum() (in module gkx.diagnostics.quasilinear_calibration) internal_vmec_backend_available() (in module gkx.geometry.backend_discovery) is_runtime_toml() (in module gkx.workflows.runtime.toml) isolated_eigenpair_observable_sensitivity_report() (in module gkx.objectives.autodiff_validation) isolated_eigenvalue_sensitivity_report() (in module gkx.objectives.autodiff_validation)

J

J() (gkx.objectives.vmec_transport.VMEXGKXTransportObjective method)

J_l_all() (in module gkx.core.velocity)

K

k_perp2() (gkx.geometry.analytic.SAlphaGeometry method) (gkx.geometry.core.SAlphaGeometry method) KBMBaseCase (class in gkx.config) KineticElectronModelConfig (class in gkx.config)

KrylovConfig (class in gkx.solvers.linear.krylov) kx_effective() (gkx.geometry.analytic.SAlphaGeometry method) (gkx.geometry.core.SAlphaGeometry method) ky_scan_batches() (in module gkx.parallel) (in module gkx.parallel.batch)

L

laguerre() (in module gkx.core.velocity) laguerre_gyroaverage_neighbors() (in module gkx.core.velocity) laguerre_quadrature_count() (in module gkx.core.velocity) laguerre_transform() (in module gkx.core.velocity) late_time_window() (in module gkx.diagnostics.growth_windows) lenard_bernstein_eigenvalues() (in module gkx.operators.linear.moments) linear_metrics_gate_report() (in module gkx.diagnostics.validation_gates) linear_rhs() (in module gkx.operators.linear.rhs) linear_rhs_cached() (in module gkx.operators.linear.rhs) linear_rhs_electrostatic_slices_velocity_sharded() (in module gkx.solvers.linear.parallel) (in module gkx.solvers.linear.parallel_electrostatic) linear_rhs_electrostatic_species_hermite_sharded() (in module gkx.solvers.linear.parallel) (in module gkx.solvers.linear.parallel_streaming) linear_rhs_electrostatic_species_sharded() (in module gkx.solvers.linear.parallel) (in module gkx.solvers.linear.parallel_electrostatic) linear_rhs_jit_for_terms_impl() (in module gkx.operators.nonlinear.rhs) linear_rhs_parallel_cached() (in module gkx.solvers.linear.parallel)

linear_rhs_streaming_electrostatic_velocity_sharded() (in module gkx.solvers.linear.parallel) (in module gkx.solvers.linear.parallel_streaming) linear_rhs_streaming_velocity_sharded() (in module gkx.solvers.linear.parallel) (in module gkx.solvers.linear.parallel_streaming) linear_solver_geometry_gradient_report() (in module gkx.objectives.gradient_gates) load_diagnostic_time_series() (in module gkx.artifacts.io) load_eigenfunction_reference_bundle() (in module gkx.diagnostics.modes) load_imported_geometry_netcdf() (in module gkx.geometry.core) (in module gkx.geometry.flux_tube) load_json_artifact() (in module gkx.diagnostics.nonlinear_gradient_evidence) load_netcdf_restart_state() (in module gkx.artifacts.io) (in module gkx.workflows.runtime.startup) load_runtime_command_config() (in module gkx.workflows.runtime.commands) load_runtime_from_toml() (in module gkx.workflows.runtime.toml) load_solved_vmex_case() (in module gkx.geometry.vmec_boozer_core) load_toml() (in module gkx.workflows.runtime.toml) local_stencil_nonlinear_domain_decomposed_step() (in module gkx.operators.nonlinear.domain_decomposition) local_stencil_nonlinear_domain_serial_step() (in module gkx.operators.nonlinear.domain_decomposition)

M

magnetic_vector_potential_energy() (in module gkx.diagnostics.moments) magnetic_vector_potential_energy_resolved() (in module gkx.diagnostics.moments) make_explicit_diagnostic_step() (in module gkx.solvers.nonlinear.explicit) make_imex_diagnostic_step() (in module gkx.solvers.nonlinear.imex) make_imex_nonlinear_term() (in module gkx.solvers.nonlinear.imex) make_imex_solve_step() (in module gkx.solvers.nonlinear.imex) make_nonlinear_diagnostic_tuple_fn() (in module gkx.operators.nonlinear.diagnostic_state) matched_nonlinear_transport_report() (in module gkx.diagnostics.validation_gates) matched_optimized_transport_report() (in module gkx.diagnostics.nonlinear_transport_optimization) metric_coeffs() (gkx.geometry.analytic.SAlphaGeometry method) (gkx.geometry.core.SAlphaGeometry method) MillerCoreParams (class in gkx.geometry.analytic) MillerGeometryRequest (class in gkx.geometry.miller_eik) mirror_drift_reference() (in module gkx.parallel) (in module gkx.parallel.velocity_streaming) mirror_drift_shard_map() (in module gkx.parallel) (in module gkx.parallel.velocity_streaming) mixing_length_amplitude2_jax() (in module gkx.diagnostics.quasilinear_transport) mode21_vmec_boozer_linear_frequency_gradient_report() (in module gkx.objectives.vmec_boozer_gradients) mode21_vmec_boozer_nonlinear_window_gradient_report() (in module gkx.objectives.vmec_boozer_gradients) mode21_vmec_boozer_quasilinear_gradient_report() (in module gkx.objectives.vmec_boozer_gradients) ModelConfig (class in gkx.config) ModeSelection (class in gkx.diagnostics.modes) ModeSelectionBatch (class in gkx.diagnostics.modes) module gkx.artifacts gkx.artifacts.io gkx.artifacts.nonlinear_netcdf gkx.artifacts.spectral_layout gkx.artifacts.zonal_plots gkx.config gkx.core.grid gkx.core.velocity gkx.diagnostics.growth_rates gkx.diagnostics.growth_windows gkx.diagnostics.modes gkx.diagnostics.moments gkx.diagnostics.nonlinear_gradient_evidence gkx.diagnostics.nonlinear_gradient_statistics gkx.diagnostics.nonlinear_replicates gkx.diagnostics.nonlinear_transport_optimization gkx.diagnostics.normalization gkx.diagnostics.quasilinear_calibration gkx.diagnostics.quasilinear_model_selection gkx.diagnostics.quasilinear_transport gkx.diagnostics.stellarator_transport_reports gkx.diagnostics.transport gkx.diagnostics.transport_windows gkx.diagnostics.validation_gates gkx.geometry.analytic gkx.geometry.autodiff_checks gkx.geometry.backend_discovery gkx.geometry.booz_xform_bridge gkx.geometry.core gkx.geometry.differentiable gkx.geometry.flux_tube gkx.geometry.flux_tube_contract gkx.geometry.miller_eik gkx.geometry.numerics gkx.geometry.sensitivity gkx.geometry.vmec_boozer_constants gkx.geometry.vmec_boozer_core gkx.geometry.vmec_boundary_chain gkx.geometry.vmec_eik gkx.geometry.vmec_flux_tube_reports gkx.geometry.vmec_state_sensitivity gkx.geometry.vmec_tensor_mapping gkx.objectives.autodiff_validation gkx.objectives.core gkx.objectives.eigen gkx.objectives.geometry gkx.objectives.gradient_gates gkx.objectives.portfolio gkx.objectives.portfolio_guard gkx.objectives.sampling gkx.objectives.stellarator_contracts gkx.objectives.stellarator_reduced gkx.objectives.stellarator_tables gkx.objectives.vmec_boozer gkx.objectives.vmec_boozer_fd gkx.objectives.vmec_boozer_gradients gkx.objectives.vmec_boozer_line_search gkx.objectives.vmec_candidate_admission gkx.objectives.vmec_transport gkx.objectives.vmec_transport_admission gkx.objectives.vmec_transport_branch gkx.objectives.vmec_transport_optimization gkx.objectives.zonal gkx.operators gkx.operators.collision gkx.operators.linear gkx.operators.linear.dissipation gkx.operators.linear.linked gkx.operators.linear.moments gkx.operators.linear.params gkx.operators.linear.rhs gkx.operators.nonlinear.brackets gkx.operators.nonlinear.collisions gkx.operators.nonlinear.device_z gkx.operators.nonlinear.diagnostic_state gkx.operators.nonlinear.domain_decomposition gkx.operators.nonlinear.projection gkx.operators.nonlinear.rhs gkx.operators.nonlinear.spectral_core gkx.parallel gkx.parallel.batch gkx.parallel.decomposition gkx.parallel.identity gkx.parallel.independent gkx.parallel.integrators gkx.parallel.state gkx.parallel.velocity_drive gkx.parallel.velocity_hermite gkx.parallel.velocity_plan gkx.parallel.velocity_streaming gkx.runtime gkx.solvers.linear.implicit gkx.solvers.linear.integrator_diagnostics gkx.solvers.linear.integrators gkx.solvers.linear.krylov gkx.solvers.linear.parallel gkx.solvers.linear.parallel_common gkx.solvers.linear.parallel_electrostatic gkx.solvers.linear.parallel_streaming gkx.solvers.nonlinear.diagnostics gkx.solvers.nonlinear.explicit gkx.solvers.nonlinear.imex gkx.solvers.time.diffrax_core gkx.solvers.time.diffrax_linear gkx.solvers.time.diffrax_nonlinear gkx.solvers.time.diffrax_streaming gkx.solvers.time.explicit gkx.solvers.time.explicit_cfl gkx.solvers.time.explicit_diagnostics gkx.solvers.time.explicit_steps gkx.solvers.time.runners gkx.terms.nonlinear gkx.workflows.runtime.chunks gkx.workflows.runtime.commands gkx.workflows.runtime.config gkx.workflows.runtime.diagnostic_arrays gkx.workflows.runtime.diagnostics gkx.workflows.runtime.execution gkx.workflows.runtime.initial_conditions gkx.workflows.runtime.initial_phi gkx.workflows.runtime.orchestration_artifacts gkx.workflows.runtime.orchestration_scan gkx.workflows.runtime.policies gkx.workflows.runtime.results gkx.workflows.runtime.startup gkx.workflows.runtime.toml

N

n_samples (gkx.objectives.portfolio.StellaratorObjectivePortfolioContract property) (gkx.objectives.stellarator_contracts.StellaratorITGSampleSet property) nonlinear_domain_identity_report() (in module gkx.operators.nonlinear.domain_decomposition) nonlinear_domain_parallel_identity_gate() (in module gkx.operators.nonlinear.domain_decomposition) nonlinear_domain_transport_window_identity_gate() (in module gkx.operators.nonlinear.domain_decomposition) nonlinear_em_components() (in module gkx.terms.nonlinear) nonlinear_em_contribution() (in module gkx.terms.nonlinear) nonlinear_em_term_cached_impl() (in module gkx.operators.nonlinear.rhs) nonlinear_field_phi2() (in module gkx.workflows.runtime.results) nonlinear_gradient_control_mean_gate() (in module gkx.diagnostics.nonlinear_gradient_statistics) nonlinear_gradient_variance_reduction_plan() (in module gkx.diagnostics.nonlinear_gradient_statistics) nonlinear_heat_flux_convergence_gate_report() (in module gkx.diagnostics.validation_gates) nonlinear_heat_flux_trace() (in module gkx.objectives.stellarator_reduced) nonlinear_heat_flux_window_metrics() (in module gkx.objectives.stellarator_reduced) nonlinear_replicate_spread_report() (in module gkx.diagnostics.nonlinear_replicates) nonlinear_rhs_cached_impl() (in module gkx.operators.nonlinear.rhs) nonlinear_spectral_domain_work_model() (in module gkx.operators.nonlinear.spectral_core) nonlinear_spectral_pencil_work_model() (in module gkx.operators.nonlinear.spectral_core) nonlinear_turbulence_gradient_bracket_sweep_report() (in module gkx.diagnostics.nonlinear_gradient_evidence) nonlinear_turbulence_gradient_candidate_ranking_report() (in module gkx.diagnostics.nonlinear_gradient_evidence) nonlinear_turbulence_gradient_evidence_gap_report() (in module gkx.diagnostics.nonlinear_gradient_evidence) nonlinear_turbulence_gradient_evidence_report() (in module gkx.diagnostics.nonlinear_gradient_evidence) nonlinear_turbulence_gradient_finite_difference_report() (in module gkx.diagnostics.nonlinear_gradient_evidence) (in module gkx.diagnostics.transport)

nonlinear_window_convergence_from_csv() (in module gkx.diagnostics.transport_windows) nonlinear_window_convergence_from_summary() (in module gkx.diagnostics.transport_windows) nonlinear_window_convergence_report() (in module gkx.diagnostics.transport_windows) nonlinear_window_ensemble_artifact_manifest() (in module gkx.diagnostics.nonlinear_replicates) nonlinear_window_ensemble_report() (in module gkx.diagnostics.transport_windows) nonlinear_window_gate_report() (in module gkx.diagnostics.validation_gates) nonlinear_window_stats_promotion_ready() (in module gkx.diagnostics.transport_windows) NonlinearArtifactPolicy (class in gkx.workflows.runtime.orchestration_artifacts) NonlinearCollisionSplitPolicy (class in gkx.operators.nonlinear.collisions) NonlinearDiagnosticKernels (class in gkx.operators.nonlinear.diagnostic_state) NonlinearGradientControlMeanGateConfig (class in gkx.diagnostics.nonlinear_gradient_statistics) NonlinearGradientVarianceReductionConfig (class in gkx.diagnostics.nonlinear_gradient_statistics) NonlinearReplicateSpreadConfig (class in gkx.diagnostics.nonlinear_replicates) NonlinearTurbulenceGradientBracketSweepConfig (class in gkx.diagnostics.nonlinear_gradient_evidence) NonlinearTurbulenceGradientCandidateRankingConfig (class in gkx.diagnostics.nonlinear_gradient_evidence) NonlinearTurbulenceGradientEvidenceConfig (class in gkx.diagnostics.nonlinear_gradient_evidence) NonlinearTurbulenceGradientFiniteDifferenceConfig (class in gkx.diagnostics.nonlinear_gradient_evidence) NonlinearTurbulenceGradientGapConfig (class in gkx.diagnostics.nonlinear_gradient_evidence) NonlinearWindowConvergenceConfig (class in gkx.diagnostics.transport_windows) NonlinearWindowEnsembleConfig (class in gkx.diagnostics.transport_windows) NonlinearWindowEnsembleManifestConfig (class in gkx.diagnostics.nonlinear_replicates) NormalizationContract (class in gkx.diagnostics.normalization) normalize_eigenfunction() (in module gkx.diagnostics.modes) normalize_quasilinear_channels() (in module gkx.diagnostics.quasilinear_transport)

O

objective_options() (gkx.objectives.vmec_transport.VMEXTransportObjectiveConfig method) objective_portfolio_sensitivity_report() (in module gkx.objectives.portfolio) objective_weights() (gkx.objectives.zonal.ZonalFlowObjectiveConfig method) observable_gradient_validation_report() (in module gkx.geometry.autodiff_checks) (in module gkx.geometry.differentiable) observed_order_gate_report() (in module gkx.diagnostics.validation_gates)

omega_d() (gkx.geometry.analytic.SAlphaGeometry method) (gkx.geometry.core.SAlphaGeometry method) omega_d_scale (gkx.diagnostics.normalization.NormalizationContract attribute) omega_star_scale (gkx.diagnostics.normalization.NormalizationContract attribute) optimization_artifact_reduction_scope() (in module gkx.diagnostics.nonlinear_transport_optimization) optimized_equilibrium_transport_report() (in module gkx.diagnostics.nonlinear_transport_optimization)

P

pad_to_multiple() (in module gkx.parallel) (in module gkx.parallel.batch) parallel_identity_report() (in module gkx.parallel) (in module gkx.parallel.identity) parallel_momentum (gkx.operators.linear.dissipation.CollisionInvariantRates attribute) ParallelIdentityReport (class in gkx.parallel) (class in gkx.parallel.identity) parameters (gkx.operators.collision.CollisionContext attribute) particle_flux_channel_resolved_species() (in module gkx.diagnostics.moments) particle_flux_channel_species() (in module gkx.diagnostics.transport) particle_flux_resolved_species() (in module gkx.diagnostics.moments) particle_flux_species() (in module gkx.diagnostics.transport) particle_flux_total() (in module gkx.diagnostics.transport) periodic_streaming_reference() (in module gkx.parallel) (in module gkx.parallel.velocity_streaming) periodic_streaming_shard_map() (in module gkx.parallel) (in module gkx.parallel.velocity_streaming) phase (gkx.operators.nonlinear.projection.ShearingCoordinateUpdate attribute) phase_align_eigenfunction() (in module gkx.diagnostics.modes) phi2_resolved() (in module gkx.diagnostics.moments) phi_norm2() (in module gkx.diagnostics.quasilinear_transport) placeholder_nonlinear_contribution() (in module gkx.terms.nonlinear)

plot_saved_output_command() (in module gkx.workflows.runtime.commands) portfolio_objective_weight_vector() (in module gkx.objectives.portfolio) portfolio_sample_weight_tensor() (in module gkx.objectives.portfolio) prepare_electrostatic_species_hermite_state() (in module gkx.solvers.linear.parallel) (in module gkx.solvers.linear.parallel_streaming) prepare_electrostatic_species_inputs() (in module gkx.solvers.linear.parallel) (in module gkx.solvers.linear.parallel_electrostatic) prepare_explicit_nonlinear_diagnostics_impl() (in module gkx.solvers.nonlinear.diagnostics) PreparedExplicitNonlinearDiagnostics (class in gkx.solvers.nonlinear.diagnostics) prewarm_vmec_boozer_equal_arc_cache() (in module gkx.geometry.vmec_boozer_constants) (in module gkx.geometry.vmec_boozer_core) print_linear_run_header() (in module gkx.workflows.runtime.commands) print_nonlinear_command_outputs() (in module gkx.workflows.runtime.orchestration_artifacts) print_nonlinear_run_header() (in module gkx.workflows.runtime.commands) print_nonlinear_run_summary() (in module gkx.workflows.runtime.commands) print_saved_paths() (in module gkx.workflows.runtime.orchestration_artifacts) production_independent_batching (gkx.parallel.decomposition.DecompositionContract property) (gkx.parallel.DecompositionContract property) production_nonlinear_optimization_guard_report() (in module gkx.diagnostics.nonlinear_transport_optimization) ProductionNonlinearOptimizationGuardConfig (class in gkx.diagnostics.nonlinear_transport_optimization) projected_line_search_input_manifest() (in module gkx.objectives.vmec_transport_optimization) ProjectedLineSearchPolicy (class in gkx.objectives.vmec_transport_optimization)

Q

qa_max_mode1_observables() (in module gkx.objectives.stellarator_reduced) qa_observable_vector() (in module gkx.objectives.stellarator_reduced) quasilinear_calibration_report() (in module gkx.diagnostics.quasilinear_calibration)

quasilinear_feature_objective() (in module gkx.diagnostics.quasilinear_transport) QuasilinearCalibrationPoint (class in gkx.diagnostics.quasilinear_calibration) QuasilinearTransportResult (class in gkx.diagnostics.quasilinear_transport) quasineutrality_phi() (in module gkx.operators.linear.moments)

R

real_fft_mesh() (in module gkx.core.grid) real_fft_ordered_kx() (in module gkx.core.grid) real_fft_unique_ky() (in module gkx.core.grid) reconstruct_serial() (in module gkx.parallel) (in module gkx.parallel.decomposition) ReconstructionIdentityReport (class in gkx.parallel) (class in gkx.parallel.decomposition) reduced_artifact_scope_report() (in module gkx.diagnostics.nonlinear_transport_optimization) reduced_portfolio_artifact_guard_report() (in module gkx.objectives.portfolio_guard) ReducedPortfolioArtifactGuardConfig (class in gkx.objectives.portfolio_guard) refit_runtime_linear_trajectory() (in module gkx.workflows.runtime.diagnostics) replicated_transport_ensemble_report() (in module gkx.diagnostics.nonlinear_transport_optimization) resolve_cfl_fac() (in module gkx.config) resolve_nonlinear_artifact_policy() (in module gkx.workflows.runtime.orchestration_artifacts) resolve_runtime_path() (in module gkx.workflows.runtime.toml) resolve_state_sharding() (in module gkx.parallel) (in module gkx.parallel.state) resolve_vmex_case_input_path() (in module gkx.geometry.vmec_boozer_core) rho_star (gkx.diagnostics.normalization.NormalizationContract attribute) row_shape (gkx.objectives.portfolio.StellaratorObjectivePortfolioContract property) run() (gkx.solvers.nonlinear.diagnostics.PreparedExplicitNonlinearDiagnostics method) run_adaptive_runtime_chunk_loop() (in module gkx.workflows.runtime.chunks) run_arrays() (gkx.solvers.nonlinear.diagnostics.PreparedExplicitNonlinearDiagnostics method) run_explicit_diagnostic_scan() (in module gkx.solvers.nonlinear.explicit) run_imex_diagnostic_scan() (in module gkx.solvers.nonlinear.imex) run_linear_case() (in module gkx.workflows.runtime.commands) run_nonlinear_case() (in module gkx.workflows.runtime.commands) run_runtime_linear() (in module gkx.runtime) run_runtime_linear_command() (in module gkx.workflows.runtime.commands) run_runtime_linear_impl() (in module gkx.workflows.runtime.execution) run_runtime_nonlinear() (in module gkx.runtime) run_runtime_nonlinear_artifact_handoff() (in module gkx.workflows.runtime.orchestration_artifacts) run_runtime_nonlinear_command() (in module gkx.workflows.runtime.commands) run_runtime_nonlinear_impl() (in module gkx.workflows.runtime.execution) run_runtime_parameter_scan() (in module gkx.workflows.runtime.orchestration_scan) run_runtime_scan() (in module gkx.runtime)

run_runtime_scan_batch() (in module gkx.workflows.runtime.orchestration_scan) run_runtime_scan_ky_task() (in module gkx.workflows.runtime.orchestration_scan) run_runtime_scan_orchestration() (in module gkx.workflows.runtime.orchestration_scan) runtime_geometry_config_for_builder() (in module gkx.workflows.runtime.startup) runtime_output_path() (in module gkx.workflows.runtime.commands) RuntimeArtifactHandoffDeps (class in gkx.workflows.runtime.orchestration_artifacts) RuntimeCaseDeps (class in gkx.workflows.runtime.commands) RuntimeCollisionConfig (class in gkx.workflows.runtime.config) RuntimeCommandDeps (class in gkx.workflows.runtime.commands) RuntimeConfig (class in gkx.workflows.runtime.config) RuntimeExpertConfig (class in gkx.workflows.runtime.config) RuntimeIndependentParallelPlan (class in gkx.runtime) (class in gkx.workflows.runtime.policies) RuntimeLinearCommandOptions (class in gkx.workflows.runtime.commands) RuntimeLinearDispatchDeps (class in gkx.workflows.runtime.execution) RuntimeLinearFitResult (class in gkx.workflows.runtime.diagnostics) RuntimeLinearResult (class in gkx.runtime) (class in gkx.workflows.runtime.results) RuntimeLinearScanResult (class in gkx.runtime) (class in gkx.workflows.runtime.results) RuntimeNonlinearCommandOptions (class in gkx.workflows.runtime.commands) RuntimeNonlinearDispatchDeps (class in gkx.workflows.runtime.execution) RuntimeNonlinearResult (class in gkx.runtime) (class in gkx.workflows.runtime.results) RuntimeNormalizationConfig (class in gkx.workflows.runtime.config) RuntimeOutputConfig (class in gkx.workflows.runtime.config) RuntimeParallelConfig (class in gkx.workflows.runtime.config) RuntimeParameterScanResult (class in gkx.workflows.runtime.results) RuntimePhysicsConfig (class in gkx.workflows.runtime.config) RuntimeProgressSnapshot (class in gkx.workflows.runtime.chunks) RuntimeQuasilinearConfig (class in gkx.workflows.runtime.config) RuntimeQuasilinearFinalizationDeps (class in gkx.workflows.runtime.diagnostics) RuntimeScanBatchDeps (class in gkx.workflows.runtime.orchestration_scan) RuntimeScanCommandOptions (class in gkx.workflows.runtime.commands) RuntimeScanDeps (class in gkx.workflows.runtime.orchestration_scan) RuntimeSpeciesConfig (class in gkx.workflows.runtime.config) RuntimeTermsConfig (class in gkx.workflows.runtime.config)

S

SAlphaGeometry (class in gkx.geometry.analytic) (class in gkx.geometry.core) sample_flux_tube_geometry() (in module gkx.geometry.core) (in module gkx.geometry.flux_tube) sample_shape (gkx.objectives.portfolio.StellaratorObjectivePortfolioContract property) saturated_flux_from_linear_weight() (in module gkx.diagnostics.quasilinear_transport) saturation_amplitude2() (in module gkx.diagnostics.quasilinear_transport) save_eigenfunction_reference_bundle() (in module gkx.diagnostics.modes) ScalarGateResult (class in gkx.diagnostics.validation_gates) scan_runtime_linear_command() (in module gkx.workflows.runtime.commands) select_admitted_transport_candidate() (in module gkx.objectives.vmec_transport_admission) select_fit_window() (in module gkx.diagnostics.growth_rates) (in module gkx.diagnostics.growth_windows) select_fit_window_loglinear() (in module gkx.diagnostics.growth_rates) (in module gkx.diagnostics.growth_windows) select_ky_grid() (in module gkx.core.grid) select_ky_index() (in module gkx.diagnostics.modes) select_projected_line_search_candidate() (in module gkx.objectives.vmec_transport_optimization) select_real_fft_ky_grid() (in module gkx.core.grid) serial_reconstruction_identity_report() (in module gkx.parallel) (in module gkx.parallel.decomposition) shape_aware_power_law_objective() (in module gkx.diagnostics.quasilinear_transport) shard_sequence() (in module gkx.parallel) (in module gkx.parallel.decomposition) ShardAssignment (class in gkx.parallel) (class in gkx.parallel.decomposition) ShearingCoordinateUpdate (class in gkx.operators.nonlinear.projection) shift_axis() (in module gkx.operators.linear.moments) should_show_progress() (in module gkx.workflows.runtime.commands) single_precision_factorial() (in module gkx.core.velocity) size (gkx.parallel.decomposition.ShardAssignment property) (gkx.parallel.ShardAssignment property) SlabGeometry (class in gkx.geometry.analytic) (class in gkx.geometry.core) slice_runtime_diagnostics() (in module gkx.workflows.runtime.diagnostic_arrays) smooth_positive() (in module gkx.objectives.stellarator_reduced)

solve_imex_step() (in module gkx.solvers.nonlinear.imex) solver_grid_options_from_ky_values() (in module gkx.objectives.sampling) solver_growth_rate_from_geometry() (in module gkx.objectives.core) solver_linear_operator_matrix_from_geometry() (in module gkx.objectives.core) solver_objective_branch_gradient_report() (in module gkx.objectives.gradient_gates) solver_objective_vector_from_geometry() (in module gkx.objectives.core) solver_ready_geometry_mapping() (in module gkx.objectives.geometry) solver_scalar_objective_from_vector() (in module gkx.objectives.core) sparse_descent_direction_from_gradient_report() (in module gkx.objectives.vmec_transport_optimization) Species (class in gkx.operators.linear.params) spectral_phi_weights() (in module gkx.diagnostics.quasilinear_transport) SpectralGrid (class in gkx.core.grid) split_evenly() (in module gkx.parallel) (in module gkx.parallel.batch) split_step() (gkx.operators.collision.SplitCollisionOperator method) SplitCollisionOperator (class in gkx.operators.collision) state (gkx.operators.nonlinear.projection.ShearingCoordinateUpdate attribute) stellarator_itg_density_gradient_scan() (in module gkx.objectives.stellarator_tables) stellarator_itg_objective() (in module gkx.objectives.stellarator_reduced) stellarator_itg_objective_residual_names() (in module gkx.objectives.stellarator_reduced) stellarator_itg_objective_residual_vector() (in module gkx.objectives.stellarator_reduced) stellarator_itg_portfolio_gate_payload() (in module gkx.objectives.stellarator_tables) stellarator_itg_portfolio_sensitivity_report() (in module gkx.objectives.stellarator_tables) stellarator_itg_reduced_portfolio_objective() (in module gkx.objectives.stellarator_tables) stellarator_itg_residual_sensitivity_report() (in module gkx.objectives.stellarator_reduced) stellarator_itg_sample_objective_table() (in module gkx.objectives.stellarator_tables) stellarator_itg_vmec_boozer_portfolio_objective_from_state() (in module gkx.objectives.stellarator_tables) stellarator_itg_vmec_boozer_sample_objective_table_from_state() (in module gkx.objectives.stellarator_tables) StellaratorITGOptimizationConfig (class in gkx.objectives.stellarator_contracts) StellaratorITGOptimizationResult (class in gkx.objectives.stellarator_contracts) StellaratorITGSampleSet (class in gkx.objectives.stellarator_contracts) StellaratorObjectivePortfolioContract (class in gkx.objectives.portfolio) streaming_term() (in module gkx.operators.linear.moments) stride_runtime_diagnostics() (in module gkx.workflows.runtime.diagnostic_arrays) sum_Jl2() (in module gkx.core.velocity) summarize_window_evidence() (in module gkx.diagnostics.nonlinear_gradient_evidence) (in module gkx.diagnostics.nonlinear_replicates)

T

thermal_energy (gkx.operators.linear.dissipation.CollisionInvariantRates attribute) TimeConfig (class in gkx.config) tiny_differentiable_objective_gradient_report() (in module gkx.objectives.geometry) to_dict() (gkx.diagnostics.quasilinear_transport.QuasilinearTransportResult method) (gkx.objectives.portfolio.StellaratorObjectivePortfolioContract method) (gkx.objectives.portfolio_guard.ReducedPortfolioArtifactGuardConfig method) (gkx.objectives.stellarator_contracts.StellaratorITGOptimizationResult method) (gkx.objectives.stellarator_contracts.StellaratorITGSampleSet method) (gkx.objectives.vmec_transport_admission.VMEXNonlinearAuditPolicy method) (gkx.objectives.vmec_transport_admission.VMEXNonlinearCampaignPolicy method) (gkx.objectives.vmec_transport_admission.VMEXReducedPrelaunchPolicy method) (gkx.objectives.vmec_transport_admission.VMEXTransportAdmissionPolicy method) (gkx.objectives.vmec_transport_optimization.ProjectedLineSearchPolicy method) (gkx.objectives.zonal.ZonalFlowObjectiveConfig method) (gkx.parallel.decomposition.DecompositionContract method) (gkx.parallel.decomposition.ReconstructionIdentityReport method) (gkx.parallel.decomposition.ShardAssignment method) (gkx.parallel.DecompositionContract method) (gkx.parallel.identity.ParallelIdentityReport method) (gkx.parallel.independent.IndependentEnsembleProvenanceReport method) (gkx.parallel.independent.IndependentWorkerMetadata method) (gkx.parallel.IndependentEnsembleProvenanceReport method) (gkx.parallel.IndependentWorkerMetadata method) (gkx.parallel.ParallelIdentityReport method) (gkx.parallel.ReconstructionIdentityReport method) (gkx.parallel.ShardAssignment method) (gkx.runtime.RuntimeIndependentParallelPlan method) (gkx.workflows.runtime.policies.RuntimeIndependentParallelPlan method)

toml_shorthand_command() (in module gkx.workflows.runtime.toml) transport_objective_sample_summary() (in module gkx.objectives.vmec_transport_admission) trim_terminal_theta_point() (gkx.geometry.core.FluxTubeGeometryData method) (gkx.geometry.flux_tube.FluxTubeGeometryData method) truncate_runtime_diagnostics() (in module gkx.workflows.runtime.diagnostic_arrays) turbulent_heating_resolved_species() (in module gkx.diagnostics.moments) turbulent_heating_species() (in module gkx.diagnostics.transport) turbulent_heating_total() (in module gkx.diagnostics.transport) twist_shift_params() (in module gkx.geometry.core) twothirds_mask() (in module gkx.core.grid)

V

validate() (gkx.diagnostics.nonlinear_transport_optimization.ProductionNonlinearOptimizationGuardConfig method) validate_finite_array() (in module gkx.artifacts) (in module gkx.artifacts.io) validate_finite_runtime_diagnostics() (in module gkx.workflows.runtime.diagnostic_arrays) validate_finite_runtime_result() (in module gkx.artifacts) (in module gkx.artifacts.io) validate_objective_portfolio_contract() (in module gkx.objectives.portfolio) velocity_field_reduce_reference() (in module gkx.parallel) (in module gkx.parallel.velocity_hermite) velocity_field_reduce_shard_map() (in module gkx.parallel) (in module gkx.parallel.velocity_hermite) VelocityShardingPlan (class in gkx.parallel) (class in gkx.parallel.velocity_plan) vmec_boozer_aggregate_line_search_holdout_report() (in module gkx.objectives.vmec_boozer_line_search) vmec_boozer_aggregate_scalar_objective_finite_difference_report() (in module gkx.objectives.vmec_boozer_fd) vmec_boozer_aggregate_scalar_objective_from_state() (in module gkx.objectives.vmec_boozer) vmec_boozer_aggregate_scalar_objective_line_search_report() (in module gkx.objectives.vmec_boozer_line_search) vmec_boozer_scalar_objective_finite_difference_report() (in module gkx.objectives.vmec_boozer_fd) vmec_boozer_scalar_objective_from_state() (in module gkx.objectives.vmec_boozer) vmec_boozer_scalar_objective_line_search_report() (in module gkx.objectives.vmec_boozer_line_search) vmec_boozer_solver_objective_table_from_state() (in module gkx.objectives.vmec_boozer) vmec_boozer_solver_objective_table_with_metadata_from_state() (in module gkx.objectives.vmec_boozer) vmec_boozer_solver_objective_vector_from_state() (in module gkx.objectives.vmec_boozer) vmec_boundary_aspect_sensitivity_report() (in module gkx.geometry.booz_xform_bridge) (in module gkx.geometry.differentiable) vmec_field_line_tensor_observable_names() (in module gkx.geometry.differentiable) (in module gkx.geometry.flux_tube_contract)

vmec_metric_tensor_observable_names() (in module gkx.geometry.differentiable) (in module gkx.geometry.flux_tube_contract) VmecGeometryRequest (class in gkx.geometry.vmec_eik) vmex_boozer_equal_arc_core_profiles_from_state() (in module gkx.geometry.differentiable) (in module gkx.geometry.vmec_boozer_core) vmex_boozer_flux_tube_sensitivity_report() (in module gkx.geometry.differentiable) (in module gkx.geometry.vmec_state_sensitivity) vmex_field_line_tensor_sensitivity_report() (in module gkx.geometry.differentiable) (in module gkx.geometry.vmec_state_sensitivity) vmex_flux_tube_array_parity_report() (in module gkx.geometry.differentiable) (in module gkx.geometry.vmec_flux_tube_reports) vmex_flux_tube_mapping_from_state() (in module gkx.geometry.differentiable) (in module gkx.geometry.vmec_tensor_mapping) vmex_flux_tube_sensitivity_report() (in module gkx.geometry.differentiable) (in module gkx.geometry.vmec_flux_tube_reports) vmex_metric_tensor_sensitivity_report() (in module gkx.geometry.differentiable) (in module gkx.geometry.vmec_state_sensitivity) vmex_transport_growth_branch_locality_report_from_states() (in module gkx.objectives.vmec_transport_branch) vmex_transport_objective_from_state() (in module gkx.objectives.vmec_transport) VMEXGKXTransportObjective (class in gkx.objectives.vmec_transport) VMEXNonlinearAuditPolicy (class in gkx.objectives.vmec_transport_admission) VMEXNonlinearCampaignPolicy (class in gkx.objectives.vmec_transport_admission) VMEXReducedPrelaunchPolicy (class in gkx.objectives.vmec_transport_admission) VMEXTransportAdmissionPolicy (class in gkx.objectives.vmec_transport_admission) VMEXTransportObjectiveConfig (class in gkx.objectives.vmec_transport)

W

windowed_growth_rate_from_omega_series() (in module gkx.diagnostics.growth_rates) with_kind_defaults() (gkx.objectives.stellarator_contracts.StellaratorITGOptimizationConfig method) write_boundary_transport_gradient_report() (in module gkx.objectives.vmec_transport_optimization) write_command_outputs() (in module gkx.workflows.runtime.orchestration_artifacts) write_linear_runtime_command_outputs() (in module gkx.workflows.runtime.orchestration_artifacts) write_netcdf_restart_state() (in module gkx.artifacts.io) write_quasilinear_artifacts() (in module gkx.artifacts) (in module gkx.artifacts.io)

write_quasilinear_calibration_report() (in module gkx.diagnostics.quasilinear_calibration) write_runtime_linear_artifacts() (in module gkx.artifacts) (in module gkx.artifacts.io) write_runtime_linear_scan_artifacts() (in module gkx.artifacts) (in module gkx.artifacts.io) write_runtime_nonlinear_table_artifacts() (in module gkx.artifacts) (in module gkx.artifacts.io) write_scan_runtime_command_outputs() (in module gkx.workflows.runtime.orchestration_artifacts)

Z

zero_shear_enabled() (in module gkx.geometry.analytic) (in module gkx.geometry.core) zonal_flow_objective_artifact_from_records() (in module gkx.objectives.zonal) zonal_flow_objective_rows() (in module gkx.objectives.zonal) zonal_flow_objective_sensitivity_report() (in module gkx.objectives.zonal) zonal_flow_reduced_objective() (in module gkx.objectives.zonal)

zonal_flow_response_figure() (in module gkx.artifacts.zonal_plots) zonal_phi_line_kxt() (in module gkx.diagnostics.moments) zonal_phi_mode_kxt() (in module gkx.diagnostics.moments) zonal_response_gate_report() (in module gkx.diagnostics.validation_gates) ZonalFlowObjectiveConfig (class in gkx.objectives.zonal) ZonalFlowResponseMetrics (class in gkx.diagnostics.validation_gates)