LICENSE
README.md
pyproject.toml
sfincs_jax/__init__.py
sfincs_jax/__main__.py
sfincs_jax/adaptive_maps.py
sfincs_jax/ambipolar.py
sfincs_jax/benchmark_artifact_policy.py
sfincs_jax/boozer_bc.py
sfincs_jax/classical_transport.py
sfincs_jax/cli.py
sfincs_jax/collisionless.py
sfincs_jax/collisionless_er.py
sfincs_jax/collisionless_exb.py
sfincs_jax/collisions.py
sfincs_jax/compare.py
sfincs_jax/constrained_pas_branch.py
sfincs_jax/diagnostics.py
sfincs_jax/explicit_sparse.py
sfincs_jax/fortran.py
sfincs_jax/geometry.py
sfincs_jax/grids.py
sfincs_jax/host_refinement.py
sfincs_jax/implicit_solve.py
sfincs_jax/indices.py
sfincs_jax/input_compat.py
sfincs_jax/io.py
sfincs_jax/jax_geometry_adapters.py
sfincs_jax/magnetic_drifts.py
sfincs_jax/mapped_xgrid_objectives.py
sfincs_jax/mapped_xgrid_transport_evidence.py
sfincs_jax/memory_model.py
sfincs_jax/namelist.py
sfincs_jax/pas_smoother.py
sfincs_jax/paths.py
sfincs_jax/periodic_stencil.py
sfincs_jax/petsc_binary.py
sfincs_jax/phi1_line_search.py
sfincs_jax/phi1_newton_linear.py
sfincs_jax/phi1_newton_policy.py
sfincs_jax/plotting.py
sfincs_jax/postprocess_upstream.py
sfincs_jax/profiling.py
sfincs_jax/research_lane_policy.py
sfincs_jax/residual.py
sfincs_jax/rhs1_acceptance_policy.py
sfincs_jax/rhs1_constraint0_policy.py
sfincs_jax/rhs1_device_operator.py
sfincs_jax/rhs1_handoff.py
sfincs_jax/rhs1_host_policy.py
sfincs_jax/rhs1_large_cpu_policy.py
sfincs_jax/rhs1_pas_matrixfree.py
sfincs_jax/rhs1_pas_policy.py
sfincs_jax/rhs1_post_xblock_policy.py
sfincs_jax/rhs1_preconditioner_auto_policy.py
sfincs_jax/rhs1_preconditioner_dispatch.py
sfincs_jax/rhs1_qi_coarse.py
sfincs_jax/rhs1_qi_galerkin_policy.py
sfincs_jax/rhs1_schur_policy.py
sfincs_jax/rhs1_sparse_exact_policy.py
sfincs_jax/rhs1_sparse_polish_policy.py
sfincs_jax/rhs1_sparse_rescue_policy.py
sfincs_jax/rhs1_stage2_policy.py
sfincs_jax/rhs1_strong_auto_kind.py
sfincs_jax/rhs1_strong_control.py
sfincs_jax/rhs1_strong_fallback.py
sfincs_jax/rhs1_strong_policy.py
sfincs_jax/rhs1_xblock_policy.py
sfincs_jax/rhs1_xblock_sparse_host_policy.py
sfincs_jax/scans.py
sfincs_jax/solve_mode_policy.py
sfincs_jax/solver.py
sfincs_jax/solver_path_policy.py
sfincs_jax/solver_progress.py
sfincs_jax/solver_progress_policy.py
sfincs_jax/solver_selection_policy.py
sfincs_jax/solver_state.py
sfincs_jax/solver_trace.py
sfincs_jax/sparse.py
sfincs_jax/structured_velocity.py
sfincs_jax/transport_dense_lu.py
sfincs_jax/transport_handoff_policy.py
sfincs_jax/transport_host_gmres.py
sfincs_jax/transport_matrix.py
sfincs_jax/transport_parallel_execution.py
sfincs_jax/transport_parallel_policy.py
sfincs_jax/transport_parallel_pool.py
sfincs_jax/transport_parallel_runtime.py
sfincs_jax/transport_parallel_validation.py
sfincs_jax/transport_parallel_worker.py
sfincs_jax/transport_policy.py
sfincs_jax/transport_preconditioner_dispatch.py
sfincs_jax/transport_residual_quality.py
sfincs_jax/transport_solve_policy.py
sfincs_jax/v3.py
sfincs_jax/v3_driver.py
sfincs_jax/v3_fblock.py
sfincs_jax/v3_sparse_pattern.py
sfincs_jax/v3_system.py
sfincs_jax/validation_artifacts.py
sfincs_jax/validation_figures.py
sfincs_jax/validation_math.py
sfincs_jax/verbose.py
sfincs_jax/vmec_geometry.py
sfincs_jax/vmec_wout.py
sfincs_jax/xgrid.py
sfincs_jax.egg-info/PKG-INFO
sfincs_jax.egg-info/SOURCES.txt
sfincs_jax.egg-info/dependency_links.txt
sfincs_jax.egg-info/entry_points.txt
sfincs_jax.egg-info/requires.txt
sfincs_jax.egg-info/top_level.txt
sfincs_jax/data/equilibria/hsx3free.bc
sfincs_jax/data/equilibria/w7x-sc1.bc
sfincs_jax/data/equilibria/w7x_standardConfig.bc
sfincs_jax/data/equilibria/wout_w7x_standardConfig.nc
sfincs_jax/data/equilibria/wout_w7x_standardConfig.txt
tests/test_adaptive_maps.py
tests/test_audit_fortran_jax_operator.py
tests/test_audit_suite_output_keys.py
tests/test_audit_suite_runtime_drift.py
tests/test_benchmark_artifact_policy.py
tests/test_benchmark_case_variants.py
tests/test_benchmark_doc_claims.py
tests/test_benchmark_multi_gpu_case_throughput.py
tests/test_benchmark_pas_tz_memory_fallback.py
tests/test_benchmark_rhs1_pas_matrixfree.py
tests/test_benchmark_sharded_solve_scaling.py
tests/test_benchmark_structured_solve.py
tests/test_benchmark_transport_parallel_scaling.py
tests/test_benchmark_w7x_high_nu_preconditioners.py
tests/test_cli_solve_mode.py
tests/test_cli_validation_io_fast_coverage.py
tests/test_collision_physics_gates.py
tests/test_collisionality_artifact.py
tests/test_collisionless_operator_parity.py
tests/test_compare_reference_corruption.py
tests/test_constrained_pas_branch.py
tests/test_create_production_benchmark_inputs.py
tests/test_distributed_gmres_axis.py
tests/test_er_scan_and_ambipolar.py
tests/test_er_trajectory_sweep.py
tests/test_er_trajectory_sweep_artifact.py
tests/test_er_xdot_parity.py
tests/test_er_xidot_parity.py
tests/test_example_auto_selection_paths.py
tests/test_exb_theta_parity.py
tests/test_explicit_sparse.py
tests/test_fblock_fokker_planck_matvec_parity.py
tests/test_fblock_fused_matvec.py
tests/test_fblock_pas_matvec_parity.py
tests/test_finite_beta_vmec_example.py
tests/test_fokker_planck_phi1_reduces_to_no_phi1.py
tests/test_fortran_mpi_wrapper.py
tests/test_fortran_reference_solver_options.py
tests/test_full_system_gmres_solution_parity.py
tests/test_full_system_matvec_parity.py
tests/test_full_system_newton_krylov.py
tests/test_full_system_operator_jit.py
tests/test_full_system_residual_jvp.py
tests/test_full_system_residual_parity.py
tests/test_full_system_rhs_parity.py
tests/test_generate_autodiff_sensitivity_validation.py
tests/test_generate_fortran_suite_benchmark_summary.py
tests/test_generate_high_collisionality_trend_proxy.py
tests/test_generate_readme_fast_branch_audit.py
tests/test_generate_sfincs_paper_figs.py
tests/test_generate_simakov_helander_high_nu_run_plan.py
tests/test_generate_simakov_helander_limit_audit.py
tests/test_generate_validation_dashboard.py
tests/test_generate_w7x_ambipolar_validation.py
tests/test_generate_w7x_high_nu_performance.py
tests/test_geometry_autodiff_gates.py
tests/test_geometry_grid_helper_coverage.py
tests/test_geometry_scheme11_parity.py
tests/test_getting_started_examples.py
tests/test_gpu_solver_path_artifacts.py
tests/test_grids_scheme_coverage.py
tests/test_helper_module_coverage.py
tests/test_host_refinement.py
tests/test_implicit_linear_solve_grad.py
tests/test_input_compat.py
tests/test_io_cache_helpers.py
tests/test_io_export_and_h5_coverage.py
tests/test_io_output_policy_coverage.py
tests/test_io_precompile_policy.py
tests/test_jax_geometry_adapters.py
tests/test_magnetic_drifts_parity.py
tests/test_mapped_xgrid_objectives.py
tests/test_mapped_xgrid_transport_evidence.py
tests/test_mapped_xgrid_v3.py
tests/test_memory_model.py
tests/test_monoenergetic_full_system_parity.py
tests/test_namelist.py
tests/test_optional_ecosystem_gates.py
tests/test_optional_eqx_jaxopt_scheme4_gate.py
tests/test_optional_lineax_implicit_gate.py
tests/test_output_coordinate_physics_gates.py
tests/test_output_h5_scheme11_parity.py
tests/test_output_h5_scheme1_parity.py
tests/test_output_h5_scheme2_parity.py
tests/test_output_h5_scheme4_parity.py
tests/test_output_h5_scheme4_quick2species_parity.py
tests/test_output_h5_scheme5_parity.py
tests/test_output_key_coverage_regressions.py
tests/test_package_metadata.py
tests/test_pas_collision_operator_parity.py
tests/test_pas_projection_heuristic.py
tests/test_pas_smoother.py
tests/test_periodic_stencil.py
tests/test_phi1_history_alignment.py
tests/test_phi1_line_search.py
tests/test_phi1_newton_linear.py
tests/test_phi1_newton_policy.py
tests/test_phi1_qn_lambda_block.py
tests/test_plotting.py
tests/test_policy_module_docstrings.py
tests/test_profile_write_output_trace.py
tests/test_profiling.py
tests/test_qi_seed_smoke_artifact.py
tests/test_release_gate_metadata.py
tests/test_research_lane_policy.py
tests/test_residual_jvp.py
tests/test_rhs1_acceptance_policy.py
tests/test_rhs1_constraint0_policy.py
tests/test_rhs1_device_operator.py
tests/test_rhs1_device_operator_unit.py
tests/test_rhs1_handoff.py
tests/test_rhs1_host_policy.py
tests/test_rhs1_large_cpu_policy.py
tests/test_rhs1_pas_matrixfree.py
tests/test_rhs1_pas_policy.py
tests/test_rhs1_post_xblock_policy.py
tests/test_rhs1_preconditioner_auto_policy.py
tests/test_rhs1_qi_coarse.py
tests/test_rhs1_qi_galerkin_policy.py
tests/test_rhs1_schur_policy.py
tests/test_rhs1_schwarz_heuristic.py
tests/test_rhs1_sparse_exact_policy.py
tests/test_rhs1_sparse_first_heuristic.py
tests/test_rhs1_sparse_polish_policy.py
tests/test_rhs1_sparse_rescue_policy.py
tests/test_rhs1_stage2_policy.py
tests/test_rhs1_strong_auto_kind.py
tests/test_rhs1_strong_control.py
tests/test_rhs1_strong_policy.py
tests/test_rhs1_xblock_fallback_initial_guess.py
tests/test_rhs1_xblock_policy.py
tests/test_rhs1_xblock_sparse_host_policy.py
tests/test_rhsmode1_phi1_write_output_end_to_end.py
tests/test_rhsmode1_write_output_end_to_end.py
tests/test_rhsmode1_xmg_er_xdot_coupling.py
tests/test_run_mapped_xgrid_transport_evidence.py
tests/test_run_qi_seed_robustness.py
tests/test_runtime_helper_coverage.py
tests/test_runtime_window_attempts.py
tests/test_scaled_example_suite_reference.py
tests/test_scans_progress_and_recycle.py
tests/test_schur_precond_heuristic.py
tests/test_sharded_matvec.py
tests/test_small_regularized_lstsq.py
tests/test_solve_mode_policy.py
tests/test_solver_gmres.py
tests/test_solver_heavy_helper_coverage.py
tests/test_solver_path_artifacts.py
tests/test_solver_path_policy.py
tests/test_solver_progress.py
tests/test_solver_progress_policy.py
tests/test_solver_selection_policy.py
tests/test_solver_state_history.py
tests/test_solver_trace.py
tests/test_solver_trace_output_formats.py
tests/test_sparse_assembly.py
tests/test_sparse_csr.py
tests/test_sparse_derivative_exb_magdrift.py
tests/test_sparse_exact_lu_heuristic.py
tests/test_sparse_precond_jax.py
tests/test_state_recycle_parity.py
tests/test_structured_velocity.py
tests/test_summarize_solver_paths.py
tests/test_transport_dense_lu.py
tests/test_transport_handoff_policy.py
tests/test_transport_host_gmres.py
tests/test_transport_matrix_rhsmode2_parity.py
tests/test_transport_matrix_rhsmode3_parity.py
tests/test_transport_matrix_write_output_end_to_end.py
tests/test_transport_parallel.py
tests/test_transport_parallel_execution.py
tests/test_transport_parallel_runtime.py
tests/test_transport_parallel_validation.py
tests/test_transport_policy_coverage.py
tests/test_transport_preconditioner_dispatch.py
tests/test_transport_residual_quality.py
tests/test_transport_solve_policy.py
tests/test_transport_sparse_direct.py
tests/test_u_hat_fft.py
tests/test_upstream_scanplot2_smoke.py
tests/test_utils_sfincs_jax_driver.py
tests/test_utils_smoke.py
tests/test_v3_driver_dd_reduction_coverage.py
tests/test_v3_driver_pas_precond_policy_coverage.py
tests/test_v3_driver_policy_helpers.py
tests/test_v3_driver_rhs1_dispatch_coverage.py
tests/test_v3_driver_solve_policy_coverage.py
tests/test_v3_driver_sparse_helper_coverage.py
tests/test_v3_driver_strong_fallback_coverage.py
tests/test_v3_fblock_smoke.py
tests/test_v3_geometry_scheme4.py
tests/test_v3_sparse_pattern.py
tests/test_validation_artifacts.py
tests/test_validation_deferred_lane_gates.py
tests/test_validation_figures.py
tests/test_validation_manifest_schema.py
tests/test_validation_math.py
tests/test_validation_policy_coverage.py
tests/test_vmec_jax_workflow.py
tests/test_vmec_wout_conventions.py
tests/test_write_output_return_results.py
tests/test_xblock_tz_precond_heuristic.py