LICENSE
README.md
pyproject.toml
tests/test_accepted_output_artifacts.py
tests/test_augmented_lagrangian.py
tests/test_bcovar_lambda_axis_closure.py
tests/test_bmag_parity.py
tests/test_bootstrap_current_example.py
tests/test_bootstrap_current_fixed_point.py
tests/test_bootstrap_current_fixed_point_integration_optional.py
tests/test_booz_cli.py
tests/test_booz_input.py
tests/test_boundary_eval.py
tests/test_boundary_field.py
tests/test_bsup_parity.py
tests/test_bundled_validation_asset_gates.py
tests/test_c_kernels_axisym.py
tests/test_chipf_inversion.py
tests/test_chipf_mesh_convention.py
tests/test_cli_helpers.py
tests/test_cli_vmec2000_exec.py
tests/test_compat.py
tests/test_config_edges.py
tests/test_constraint_pipeline.py
tests/test_continuation_exact_history.py
tests/test_continuation_stage_inputs.py
tests/test_converged_wout_matrix_parity.py
tests/test_converged_wout_parity_benchmark.py
tests/test_coords_kernel.py
tests/test_current_driven_iota_helper.py
tests/test_diagnostics_fieldlines.py
tests/test_diagnostics_wave3_coverage.py
tests/test_differentiation_optimization_api_fast.py
tests/test_discrete_adjoint_chunking.py
tests/test_discrete_adjoint_qh.py
tests/test_discrete_adjoint_wave6_coverage.py
tests/test_docs_release_hygiene.py
tests/test_doctor.py
tests/test_driver_api.py
tests/test_driver_api_finish_more_coverage.py
tests/test_driver_fast_reconstruction.py
tests/test_driver_finish_policy_more_coverage.py
tests/test_driver_helper_edges_wave14_coverage.py
tests/test_driver_implicit_wave11_coverage.py
tests/test_driver_policy_coverage_extra.py
tests/test_driver_policy_helpers.py
tests/test_driver_run_wave8_coverage.py
tests/test_driver_wave12_coverage.py
tests/test_driver_wave2_coverage.py
tests/test_driver_wave4_coverage.py
tests/test_driver_wave5_coverage.py
tests/test_driver_wave6_coverage.py
tests/test_driver_wout_wave9_coverage.py
tests/test_dump_helpers.py
tests/test_end_to_end_vmec_residual_gn.py
tests/test_energy_fast_helpers.py
tests/test_energy_integrals_parity.py
tests/test_equif_eqfor_parity.py
tests/test_examples_api_wave13_coverage.py
tests/test_external_fields_coils_jax.py
tests/test_external_fields_essos_adapter.py
tests/test_external_fields_mgrid_jax.py
tests/test_external_vmec_asset_manifest.py
tests/test_fast_physics_kernels.py
tests/test_fetch_assets.py
tests/test_field_identities.py
tests/test_finite_beta.py
tests/test_finite_beta_examples.py
tests/test_finite_beta_helpers_unit.py
tests/test_fixaray_trig_tables.py
tests/test_fixed_boundary_profile_summary.py
tests/test_force_norms_dynamic_parity.py
tests/test_forces_bcovar_wave12_coverage.py
tests/test_fourier_geom_fast_helpers.py
tests/test_free_boundary_additional_helpers.py
tests/test_free_boundary_adjoint_helpers_unit.py
tests/test_free_boundary_beta_response_validation.py
tests/test_free_boundary_coil_provider_forward.py
tests/test_free_boundary_coil_provider_gradients.py
tests/test_free_boundary_derivatives_public_api.py
tests/test_free_boundary_direct_coil_finite_pressure_sensitivity.py
tests/test_free_boundary_direct_coils_forward_example.py
tests/test_free_boundary_essos_coil_parity.py
tests/test_free_boundary_essos_coils_forward_example.py
tests/test_free_boundary_essos_pedagogic_examples.py
tests/test_free_boundary_fast_physics_coverage.py
tests/test_free_boundary_helper_branches.py
tests/test_free_boundary_jax_nestor_operator_cache.py
tests/test_free_boundary_qa_finite_beta_coil_optimization_smoke.py
tests/test_free_boundary_qs_coil_optimization_smoke.py
tests/test_free_boundary_vacuum_adjoint.py
tests/test_free_boundary_validation_unit.py
tests/test_free_boundary_wave2.py
tests/test_free_boundary_wp0.py
tests/test_freeb_direct_coil_matrix_benchmark.py
tests/test_freeb_scalpot_compare_parser.py
tests/test_geom_metrics.py
tests/test_getfsq_block_sums.py
tests/test_glasser_diagnostic_example.py
tests/test_glasser_resistive_interchange.py
tests/test_gpu_cpu_performance_profile.py
tests/test_grid_radial_nyquist_helpers.py
tests/test_implicit_adjoint_helpers.py
tests/test_implicit_differentiation_fast.py
tests/test_implicit_helpers.py
tests/test_implicit_more_coverage.py
tests/test_implicit_residual_adjoint_helpers.py
tests/test_implicit_sensitivity_fast_coverage.py
tests/test_implicit_wave12_coverage.py
tests/test_implicit_wave4_coverage.py
tests/test_implicit_wave5_coverage.py
tests/test_implicit_wave6_coverage.py
tests/test_implicit_wout_driver_branch_coverage.py
tests/test_init_guess.py
tests/test_init_guess_fast_helpers.py
tests/test_init_plotting_wave12_coverage.py
tests/test_jax_compile_log_summary.py
tests/test_lforbal_fast_helpers.py
tests/test_local_ci_gate.py
tests/test_minimal_seed_showcase.py
tests/test_modes_helpers.py
tests/test_multigrid_interp.py
tests/test_namelist.py
tests/test_non_solve_wave6_coverage.py
tests/test_nonaxis_exec_stage_trace_parity.py
tests/test_numerics_helper_edges.py
tests/test_optimization_auto_scalar_policy.py
tests/test_optimization_callback_trace.py
tests/test_optimization_examples.py
tests/test_optimization_fast_optimizer_methods.py
tests/test_optimization_helpers.py
tests/test_optimization_wave2_coverage.py
tests/test_optimization_wave4_coverage.py
tests/test_optimization_workflow_qi_objectives_more_coverage.py
tests/test_optimization_workflow_unit.py
tests/test_optimizer_domain_helpers.py
tests/test_packaging_metadata.py
tests/test_parity_axis_rules.py
tests/test_parity_manifest_extra.py
tests/test_parity_sweep_manifest_thresholds.py
tests/test_performance_wave13_coverage.py
tests/test_performance_wave3_coverage.py
tests/test_physics_gate_wave13_coverage.py
tests/test_physics_parity_helper_gates.py
tests/test_plotting_fast_helpers.py
tests/test_plotting_helpers.py
tests/test_plotting_unit.py
tests/test_preconditioner_1d_fast_helpers.py
tests/test_preconditioner_1d_jax_fast_helpers.py
tests/test_preconditioner_plotting_coords_wave8_coverage.py
tests/test_profile_input_examples.py
tests/test_profile_report_compare.py
tests/test_profiles_fast_helpers.py
tests/test_public_module_helpers.py
tests/test_qi_basin_promote.py
tests/test_qi_basin_survey.py
tests/test_qi_case_resolution.py
tests/test_qi_constraint_policy_scan.py
tests/test_qi_diagnostics.py
tests/test_qi_filter_search.py
tests/test_qi_landscape_scan.py
tests/test_qi_legacy.py
tests/test_qi_objective_component_report.py
tests/test_qi_optimization_context_more_coverage.py
tests/test_qi_optimization_more_coverage.py
tests/test_qi_optimization_public_helpers.py
tests/test_qi_parameter_probe_harness.py
tests/test_qi_readme_cases.py
tests/test_qi_seed_evidence_artifacts.py
tests/test_qi_seed_robustness_plan.py
tests/test_qi_seed_suitability_audit.py
tests/test_qi_stage_checkpoint.py
tests/test_qi_staged_runner.py
tests/test_qi_wout_parity.py
tests/test_qs_ess_render_smoke.py
tests/test_quasi_isodynamic.py
tests/test_quasisymmetry.py
tests/test_readme_ad_fd_evidence.py
tests/test_redl_bootstrap_simsopt_parity.py
tests/test_refactorable_seams_coverage.py
tests/test_regression_cases.py
tests/test_required_helper_coverage_margin.py
tests/test_residual_iteration_preconditioner.py
tests/test_residual_ptau.py
tests/test_residue_finite_beta_wave3.py
tests/test_residue_getfsq_parity.py
tests/test_resume_state.py
tests/test_robust_coil_perturbations.py
tests/test_runtime_diagnostics.py
tests/test_scan_helper_edge_gates.py
tests/test_scan_runner_reuse_diagnostic.py
tests/test_simsopt_optional_validation.py
tests/test_simsopt_style_profiles.py
tests/test_solve_adaptive_controls.py
tests/test_solve_additional_branch_coverage.py
tests/test_solve_additional_helpers.py
tests/test_solve_axis_helpers_more_coverage.py
tests/test_solve_branch_coverage.py
tests/test_solve_debug_dump_wave10_coverage.py
tests/test_solve_diagnostics_io.py
tests/test_solve_driver_control_fast.py
tests/test_solve_dump_helpers.py
tests/test_solve_finish_cache_more_coverage.py
tests/test_solve_force_norm_helpers.py
tests/test_solve_force_payload_helpers.py
tests/test_solve_gd_wave10_coverage.py
tests/test_solve_hotpaths.py
tests/test_solve_lbfgs_wave8_coverage.py
tests/test_solve_more_coverage.py
tests/test_solve_optimizer_helpers.py
tests/test_solve_options.py
tests/test_solve_performance_instrumentation.py
tests/test_solve_preconditioner_metric_helpers.py
tests/test_solve_preconditioner_payload_helpers.py
tests/test_solve_real_scan_wave10_coverage.py
tests/test_solve_residual_iter_config.py
tests/test_solve_residual_iter_finalize_helpers.py
tests/test_solve_residual_iter_force_cache_helpers.py
tests/test_solve_residual_iter_force_payload_helpers.py
tests/test_solve_residual_iter_geometry_helpers.py
tests/test_solve_residual_iter_helpers_wave8_coverage.py
tests/test_solve_residual_iter_mode_transform_helpers.py
tests/test_solve_residual_iter_policy.py
tests/test_solve_residual_iter_policy_gap_coverage.py
tests/test_solve_residual_iter_runtime_helpers.py
tests/test_solve_residual_iter_setup_helpers.py
tests/test_solve_residual_iter_update_helpers.py
tests/test_solve_residual_objective_helpers.py
tests/test_solve_residual_optimizer_wave8_coverage.py
tests/test_solve_runtime.py
tests/test_solve_scan_chunking.py
tests/test_solve_scan_debug_helpers.py
tests/test_solve_scan_math_helpers.py
tests/test_solve_scan_output.py
tests/test_solve_scan_output_edge_cases_more_coverage.py
tests/test_solve_scan_payload_helpers.py
tests/test_solve_scan_planning_helpers.py
tests/test_solve_scan_resume_state.py
tests/test_solve_scan_time_control.py
tests/test_solve_wave3_coverage.py
tests/test_solve_wave4_coverage.py
tests/test_solve_wave6_coverage.py
tests/test_solve_wave7_coverage.py
tests/test_source_health_diagnostics.py
tests/test_state_coords_helpers.py
tests/test_static_more_coverage.py
tests/test_step3_profiles.py
tests/test_step4_field_cartesian.py
tests/test_step4_field_energy.py
tests/test_step5_solve_lambda.py
tests/test_step6_solve_fixed_boundary.py
tests/test_step8_preconditioner.py
tests/test_step8_wout_stationarity.py
tests/test_step9_implicit_fixed_boundary.py
tests/test_step9_implicit_lambda.py
tests/test_tcon_precondn_diag.py
tests/test_validation_gates_extra.py
tests/test_visualization_vtk.py
tests/test_vmec2000_converged_parity.py
tests/test_vmec2000_exec_fast_validation.py
tests/test_vmec2000_exec_parser_more_coverage.py
tests/test_vmec2000_exec_qa_regression.py
tests/test_vmec2000_exec_qh_reactorscale_multigrid_override.py
tests/test_vmec2000_exec_threed1.py
tests/test_vmec2000_fixed_boundary_physics_gates.py
tests/test_vmec2000_python_api_smoke.py
tests/test_vmec2000_scalars_parity.py
tests/test_vmec_alias_gcon.py
tests/test_vmec_bcovar_smoke.py
tests/test_vmec_constraints_fast_helpers.py
tests/test_vmec_forces_fast_helpers.py
tests/test_vmec_forces_freeb_edge.py
tests/test_vmec_forces_more_coverage.py
tests/test_vmec_forces_synthetic_helpers.py
tests/test_vmec_jacobian_parity.py
tests/test_vmec_kernel_additional_helpers.py
tests/test_vmec_numpy_forces_cache.py
tests/test_vmec_parity_host.py
tests/test_vmec_parity_physics_fast_gates.py
tests/test_vmec_physics_helper_more_coverage.py
tests/test_vmec_realspace.py
tests/test_vmec_realspace_geom.py
tests/test_vmec_realspace_invariants.py
tests/test_vmec_residue_fast_helpers.py
tests/test_vmec_tomnsp_branch_coverage.py
tests/test_vmec_tomnsp_tables.py
tests/test_vmecpp_restart_flag.py
tests/test_wout_additional_helpers.py
tests/test_wout_bcovar_forces_extra_coverage.py
tests/test_wout_beta_eqfor_bundled_parity.py
tests/test_wout_bfield_bundled_parity.py
tests/test_wout_branch_coverage.py
tests/test_wout_chipf_bundled_parity.py
tests/test_wout_comprehensive_parity.py
tests/test_wout_contravariant_field_gate.py
tests/test_wout_current_profile.py
tests/test_wout_driver_wave10_coverage.py
tests/test_wout_env_branch_coverage.py
tests/test_wout_family_converged_quantities.py
tests/test_wout_fast_helpers.py
tests/test_wout_fixture_inventory.py
tests/test_wout_geometry_mercier_bundled_parity.py
tests/test_wout_helpers.py
tests/test_wout_io_helpers.py
tests/test_wout_lasym_bsubv_parity.py
tests/test_wout_parity_reference.py
tests/test_wout_physics_gates.py
tests/test_wout_physics_wave8_coverage.py
tests/test_wout_profile_example.py
tests/test_wout_profiles_currents_bundled_parity.py
tests/test_wout_roundtrip.py
tests/test_wout_vmecplot2_compat.py
tests/test_wout_wave2.py
tests/test_wout_wave3_coverage.py
tests/test_wout_wave4_coverage.py
tests/test_wout_wave5_coverage.py
vmec_jax/__init__.py
vmec_jax/__main__.py
vmec_jax/_compat.py
vmec_jax/_solve_runtime.py
vmec_jax/api.py
vmec_jax/bootstrap_current.py
vmec_jax/booz.py
vmec_jax/booz_input.py
vmec_jax/boundary.py
vmec_jax/cli.py
vmec_jax/config.py
vmec_jax/coords.py
vmec_jax/diagnostics.py
vmec_jax/discrete_adjoint.py
vmec_jax/doctor.py
vmec_jax/driver.py
vmec_jax/energy.py
vmec_jax/field.py
vmec_jax/fieldlines.py
vmec_jax/finite_beta.py
vmec_jax/fourier.py
vmec_jax/free_boundary.py
vmec_jax/geom.py
vmec_jax/grids.py
vmec_jax/implicit.py
vmec_jax/init_guess.py
vmec_jax/integrals.py
vmec_jax/mercier.py
vmec_jax/modes.py
vmec_jax/multigrid.py
vmec_jax/namelist.py
vmec_jax/nyquist.py
vmec_jax/optimization.py
vmec_jax/optimization_workflow.py
vmec_jax/plotting.py
vmec_jax/preconditioner_1d.py
vmec_jax/preconditioner_1d_jax.py
vmec_jax/profiles.py
vmec_jax/quasisymmetry.py
vmec_jax/radial.py
vmec_jax/redl_bootstrap.py
vmec_jax/residuals.py
vmec_jax/robust_coils.py
vmec_jax/solve.py
vmec_jax/state.py
vmec_jax/static.py
vmec_jax/visualization.py
vmec_jax/vmec2000_exec.py
vmec_jax/wout.py
vmec_jax.egg-info/PKG-INFO
vmec_jax.egg-info/SOURCES.txt
vmec_jax.egg-info/dependency_links.txt
vmec_jax.egg-info/entry_points.txt
vmec_jax.egg-info/requires.txt
vmec_jax.egg-info/top_level.txt
vmec_jax/drivers/__init__.py
vmec_jax/drivers/debug.py
vmec_jax/drivers/dynamic_scan.py
vmec_jax/drivers/flux.py
vmec_jax/drivers/interface.py
vmec_jax/drivers/lifecycle.py
vmec_jax/drivers/output.py
vmec_jax/drivers/policy.py
vmec_jax/drivers/results.py
vmec_jax/drivers/runtime.py
vmec_jax/drivers/solve.py
vmec_jax/drivers/staging.py
vmec_jax/external_fields/__init__.py
vmec_jax/external_fields/base.py
vmec_jax/external_fields/coils_jax.py
vmec_jax/external_fields/essos_adapter.py
vmec_jax/external_fields/mgrid_jax.py
vmec_jax/io/__init__.py
vmec_jax/io/wout_files/__init__.py
vmec_jax/io/wout_files/bsubs.py
vmec_jax/io/wout_files/debug.py
vmec_jax/io/wout_files/diagnostics.py
vmec_jax/io/wout_files/flux.py
vmec_jax/io/wout_files/jxbforce.py
vmec_jax/io/wout_files/mercier.py
vmec_jax/io/wout_files/minimal.py
vmec_jax/io/wout_files/netcdf.py
vmec_jax/io/wout_files/nyquist.py
vmec_jax/io/wout_files/parity.py
vmec_jax/io/wout_files/schema.py
vmec_jax/io/wout_files/state.py
vmec_jax/kernels/__init__.py
vmec_jax/kernels/bcovar.py
vmec_jax/kernels/constraints.py
vmec_jax/kernels/forces.py
vmec_jax/kernels/jacobian.py
vmec_jax/kernels/lforbal.py
vmec_jax/kernels/numpy_forces.py
vmec_jax/kernels/parity.py
vmec_jax/kernels/realspace.py
vmec_jax/kernels/residue.py
vmec_jax/kernels/tomnsp.py
vmec_jax/optimizers/__init__.py
vmec_jax/optimizers/fixed_boundary/__init__.py
vmec_jax/optimizers/fixed_boundary/exact_replay.py
vmec_jax/optimizers/fixed_boundary/finite_beta_objectives.py
vmec_jax/optimizers/fixed_boundary/gauss_newton.py
vmec_jax/optimizers/fixed_boundary/history.py
vmec_jax/optimizers/fixed_boundary/linear_guards.py
vmec_jax/optimizers/fixed_boundary/matrix_free.py
vmec_jax/optimizers/fixed_boundary/objective_terms.py
vmec_jax/optimizers/fixed_boundary/parameterization.py
vmec_jax/optimizers/fixed_boundary/profiling.py
vmec_jax/optimizers/fixed_boundary/qs_residuals.py
vmec_jax/optimizers/fixed_boundary/replay_policy.py
vmec_jax/optimizers/fixed_boundary/scalar_gradient.py
vmec_jax/optimizers/fixed_boundary/scalar_lbfgs.py
vmec_jax/optimizers/fixed_boundary/scalar_trust.py
vmec_jax/optimizers/fixed_boundary/scipy_least_squares.py
vmec_jax/optimizers/fixed_boundary/seed_inputs.py
vmec_jax/optimizers/fixed_boundary/stage_policy.py
vmec_jax/optimizers/fixed_boundary/state_cache.py
vmec_jax/optimizers/fixed_boundary/workflow_artifacts.py
vmec_jax/optimizers/fixed_boundary/workflow_outputs.py
vmec_jax/quasi_isodynamic/__init__.py
vmec_jax/quasi_isodynamic/diagnostics.py
vmec_jax/quasi_isodynamic/legacy.py
vmec_jax/quasi_isodynamic/objectives.py
vmec_jax/quasi_isodynamic/optimization.py
vmec_jax/quasi_isodynamic/optimization_terms.py
vmec_jax/resources/input.nfp4_QH_warm_start
vmec_jax/solvers/__init__.py
vmec_jax/solvers/fixed_boundary/__init__.py
vmec_jax/solvers/fixed_boundary/api.py
vmec_jax/solvers/fixed_boundary/jit_cache.py
vmec_jax/solvers/fixed_boundary/options.py
vmec_jax/solvers/fixed_boundary/performance.py
vmec_jax/solvers/fixed_boundary/profiles.py
vmec_jax/solvers/fixed_boundary/results.py
vmec_jax/solvers/fixed_boundary/adjoint/__init__.py
vmec_jax/solvers/fixed_boundary/adjoint/implicit_linear_algebra.py
vmec_jax/solvers/fixed_boundary/adjoint/replay_payload.py
vmec_jax/solvers/fixed_boundary/adjoint/replay_policy.py
vmec_jax/solvers/fixed_boundary/adjoint/residual_linear_algebra.py
vmec_jax/solvers/fixed_boundary/adjoint/strict_updates.py
vmec_jax/solvers/fixed_boundary/diagnostics/__init__.py
vmec_jax/solvers/fixed_boundary/diagnostics/axis_reset.py
vmec_jax/solvers/fixed_boundary/diagnostics/bsub.py
vmec_jax/solvers/fixed_boundary/diagnostics/first_step.py
vmec_jax/solvers/fixed_boundary/diagnostics/force.py
vmec_jax/solvers/fixed_boundary/diagnostics/hlo.py
vmec_jax/solvers/fixed_boundary/diagnostics/io.py
vmec_jax/solvers/fixed_boundary/diagnostics/lambda_debug.py
vmec_jax/solvers/fixed_boundary/diagnostics/metric.py
vmec_jax/solvers/fixed_boundary/optimization/__init__.py
vmec_jax/solvers/fixed_boundary/optimization/constraints.py
vmec_jax/solvers/fixed_boundary/optimization/energy.py
vmec_jax/solvers/fixed_boundary/optimization/gd.py
vmec_jax/solvers/fixed_boundary/optimization/gradient.py
vmec_jax/solvers/fixed_boundary/optimization/lambda_gd.py
vmec_jax/solvers/fixed_boundary/optimization/lbfgs.py
vmec_jax/solvers/fixed_boundary/optimization/quasi_newton.py
vmec_jax/solvers/fixed_boundary/optimization/residual_context.py
vmec_jax/solvers/fixed_boundary/optimization/residual_gn.py
vmec_jax/solvers/fixed_boundary/optimization/residual_lbfgs.py
vmec_jax/solvers/fixed_boundary/optimization/residual_objective.py
vmec_jax/solvers/fixed_boundary/optimization/tolerances.py
vmec_jax/solvers/fixed_boundary/preconditioning/__init__.py
vmec_jax/solvers/fixed_boundary/preconditioning/operators.py
vmec_jax/solvers/fixed_boundary/preconditioning/payload.py
vmec_jax/solvers/fixed_boundary/residual/__init__.py
vmec_jax/solvers/fixed_boundary/residual/accelerated_scan.py
vmec_jax/solvers/fixed_boundary/residual/config.py
vmec_jax/solvers/fixed_boundary/residual/finalize.py
vmec_jax/solvers/fixed_boundary/residual/force_cache.py
vmec_jax/solvers/fixed_boundary/residual/force_norms.py
vmec_jax/solvers/fixed_boundary/residual/force_payload.py
vmec_jax/solvers/fixed_boundary/residual/geometry.py
vmec_jax/solvers/fixed_boundary/residual/host_diagnostics.py
vmec_jax/solvers/fixed_boundary/residual/iteration.py
vmec_jax/solvers/fixed_boundary/residual/iteration_control.py
vmec_jax/solvers/fixed_boundary/residual/iteration_metrics.py
vmec_jax/solvers/fixed_boundary/residual/iteration_preconditioner.py
vmec_jax/solvers/fixed_boundary/residual/mode_transform.py
vmec_jax/solvers/fixed_boundary/residual/payload_blocks.py
vmec_jax/solvers/fixed_boundary/residual/policy.py
vmec_jax/solvers/fixed_boundary/residual/preconditioner_payload.py
vmec_jax/solvers/fixed_boundary/residual/ptau.py
vmec_jax/solvers/fixed_boundary/residual/runtime.py
vmec_jax/solvers/fixed_boundary/residual/scan_adapters.py
vmec_jax/solvers/fixed_boundary/residual/setup.py
vmec_jax/solvers/fixed_boundary/residual/state_setup.py
vmec_jax/solvers/fixed_boundary/residual/update.py
vmec_jax/solvers/fixed_boundary/scan/__init__.py
vmec_jax/solvers/fixed_boundary/scan/controller.py
vmec_jax/solvers/fixed_boundary/scan/debug.py
vmec_jax/solvers/fixed_boundary/scan/math.py
vmec_jax/solvers/fixed_boundary/scan/output.py
vmec_jax/solvers/fixed_boundary/scan/payload.py
vmec_jax/solvers/fixed_boundary/scan/planning.py
vmec_jax/solvers/fixed_boundary/scan/resume.py
vmec_jax/solvers/fixed_boundary/scan/runtime.py
vmec_jax/solvers/fixed_boundary/scan/time_control.py
vmec_jax/solvers/free_boundary/__init__.py
vmec_jax/solvers/free_boundary/axis_current.py
vmec_jax/solvers/free_boundary/boundary_fields.py
vmec_jax/solvers/free_boundary/coil_optimization.py
vmec_jax/solvers/free_boundary/control.py
vmec_jax/solvers/free_boundary/derivatives.py
vmec_jax/solvers/free_boundary/diagnostics.py
vmec_jax/solvers/free_boundary/jax_nestor_operator.py
vmec_jax/solvers/free_boundary/mgrid.py
vmec_jax/solvers/free_boundary/types.py
vmec_jax/solvers/free_boundary/validation.py
vmec_jax/solvers/free_boundary/adjoint/__init__.py
vmec_jax/solvers/free_boundary/adjoint/boundary_replay.py
vmec_jax/solvers/free_boundary/adjoint/branch_local.py
vmec_jax/solvers/free_boundary/adjoint/branch_local_derivatives.py
vmec_jax/solvers/free_boundary/adjoint/branch_metadata.py
vmec_jax/solvers/free_boundary/adjoint/complete_solve_reports.py
vmec_jax/solvers/free_boundary/adjoint/controller.py
vmec_jax/solvers/free_boundary/adjoint/controller_checks.py
vmec_jax/solvers/free_boundary/adjoint/controller_replay.py
vmec_jax/solvers/free_boundary/adjoint/custom_vjp.py
vmec_jax/solvers/free_boundary/adjoint/dense.py
vmec_jax/solvers/free_boundary/adjoint/direct_coil_replay.py
vmec_jax/solvers/free_boundary/adjoint/gate_reports.py
vmec_jax/solvers/free_boundary/adjoint/mode_operator.py
vmec_jax/solvers/free_boundary/adjoint/mode_solve.py
vmec_jax/solvers/free_boundary/adjoint/objectives.py
vmec_jax/solvers/free_boundary/adjoint/projected_modes.py
vmec_jax/solvers/free_boundary/adjoint/pytrees.py
vmec_jax/solvers/free_boundary/adjoint/replay_context.py
vmec_jax/solvers/free_boundary/adjoint/replay_plan.py
vmec_jax/solvers/free_boundary/adjoint/runtime.py
vmec_jax/solvers/free_boundary/adjoint/trace_controls.py
vmec_jax/solvers/free_boundary/adjoint/trace_fingerprint.py
vmec_jax/solvers/free_boundary/adjoint/trace_metadata.py
vmec_jax/solvers/free_boundary/adjoint/trace_stack.py
vmec_jax/solvers/free_boundary/adjoint/vmec_nestor.py