LICENSE
README.md
pyproject.toml
torax/__init__.py
torax/conftest.py
torax/constants.py
torax/math_utils.py
torax/output.py
torax/pedestal.py
torax/run_simulation_main.py
torax/sources.py
torax/torax_pydantic.py
torax/transport.py
torax.egg-info/PKG-INFO
torax.egg-info/SOURCES.txt
torax.egg-info/dependency_links.txt
torax.egg-info/entry_points.txt
torax.egg-info/requires.txt
torax.egg-info/top_level.txt
torax/_src/array_typing.py
torax/_src/constants.py
torax/_src/interpolated_param.py
torax/_src/math_utils.py
torax/_src/model_state.py
torax/_src/models.py
torax/_src/path_utils.py
torax/_src/simulation_app.py
torax/_src/static_dataclass.py
torax/_src/tridiagonal.py
torax/_src/version.py
torax/_src/config/__init__.py
torax/_src/config/build_runtime_params.py
torax/_src/config/config_loader.py
torax/_src/config/numerics.py
torax/_src/config/runtime_params.py
torax/_src/config/runtime_validation_utils.py
torax/_src/core_profiles/convertors.py
torax/_src/core_profiles/core_profiles.py
torax/_src/core_profiles/getters.py
torax/_src/core_profiles/initialization.py
torax/_src/core_profiles/ion_composition.py
torax/_src/core_profiles/kinetic_profiles.py
torax/_src/core_profiles/poloidal_flux_profiles.py
torax/_src/core_profiles/profile_conditions.py
torax/_src/core_profiles/runtime_params.py
torax/_src/core_profiles/solver_profiles.py
torax/_src/core_profiles/solver_state.py
torax/_src/core_profiles/updaters.py
torax/_src/core_profiles/plasma_composition/electron_density_ratios.py
torax/_src/core_profiles/plasma_composition/electron_density_ratios_zeff.py
torax/_src/core_profiles/plasma_composition/impurity_fractions.py
torax/_src/core_profiles/plasma_composition/ion_mixture.py
torax/_src/core_profiles/plasma_composition/plasma_composition.py
torax/_src/edge/__init__.py
torax/_src/edge/base.py
torax/_src/edge/config.py
torax/_src/edge/edge_state.py
torax/_src/edge/register_model.py
torax/_src/edge/runtime_params.py
torax/_src/edge/updaters.py
torax/_src/edge/extended_lengyel/__init__.py
torax/_src/edge/extended_lengyel/collisional_radiative_models.py
torax/_src/edge/extended_lengyel/config.py
torax/_src/edge/extended_lengyel/divertor_sol_1d.py
torax/_src/edge/extended_lengyel/extended_lengyel_defaults.py
torax/_src/edge/extended_lengyel/extended_lengyel_enums.py
torax/_src/edge/extended_lengyel/extended_lengyel_formulas.py
torax/_src/edge/extended_lengyel/extended_lengyel_model.py
torax/_src/edge/extended_lengyel/extended_lengyel_solvers.py
torax/_src/edge/extended_lengyel/extended_lengyel_standalone.py
torax/_src/edge/extended_lengyel/mavrin_2017_charge_states_data.py
torax/_src/fvm/__init__.py
torax/_src/fvm/block_1d_coeffs.py
torax/_src/fvm/calc_coeffs.py
torax/_src/fvm/cell_variable.py
torax/_src/fvm/convection_terms.py
torax/_src/fvm/diffusion_terms.py
torax/_src/fvm/discrete_system.py
torax/_src/fvm/enums.py
torax/_src/fvm/fvm_conversions.py
torax/_src/fvm/implicit_solve_block.py
torax/_src/fvm/newton_raphson_solve_block.py
torax/_src/fvm/optimizer_solve_block.py
torax/_src/fvm/residual_and_loss.py
torax/_src/geometry/__init__.py
torax/_src/geometry/base.py
torax/_src/geometry/chease.py
torax/_src/geometry/circular_geometry.py
torax/_src/geometry/config.py
torax/_src/geometry/eqdsk.py
torax/_src/geometry/geometry.py
torax/_src/geometry/geometry_loader.py
torax/_src/geometry/geometry_model.py
torax/_src/geometry/geometry_state.py
torax/_src/geometry/get_example_L_LY_data.py
torax/_src/geometry/imas.py
torax/_src/geometry/meq.py
torax/_src/geometry/runtime_params.py
torax/_src/geometry/standard_geometry.py
torax/_src/geometry/tokamaker.py
torax/_src/geometry/trapped_fraction.py
torax/_src/imas_tools/sources_mapping.py
torax/_src/imas_tools/input/__init__.py
torax/_src/imas_tools/input/core_profiles.py
torax/_src/imas_tools/input/core_sources.py
torax/_src/imas_tools/input/equilibrium.py
torax/_src/imas_tools/input/loader.py
torax/_src/imas_tools/input/validation.py
torax/_src/imas_tools/output/__init__.py
torax/_src/imas_tools/output/core_profiles.py
torax/_src/imas_tools/output/core_sources.py
torax/_src/imas_tools/output/equilibrium.py
torax/_src/internal_boundary_conditions/__init__.py
torax/_src/internal_boundary_conditions/base_model.py
torax/_src/internal_boundary_conditions/beta_poloidal_prime.py
torax/_src/internal_boundary_conditions/builder.py
torax/_src/internal_boundary_conditions/config.py
torax/_src/internal_boundary_conditions/internal_boundary_conditions.py
torax/_src/internal_boundary_conditions/no_ibc.py
torax/_src/internal_boundary_conditions/prescribed.py
torax/_src/internal_boundary_conditions/runtime_params.py
torax/_src/jax_utils/__init__.py
torax/_src/jax_utils/common.py
torax/_src/jax_utils/matrix_inverse.py
torax/_src/jax_utils/while_loop_bounded.py
torax/_src/mhd/base.py
torax/_src/mhd/config.py
torax/_src/mhd/runtime_params.py
torax/_src/mhd/sawtooth/config.py
torax/_src/mhd/sawtooth/flatten_profile.py
torax/_src/mhd/sawtooth/redistribution_base.py
torax/_src/mhd/sawtooth/runtime_params.py
torax/_src/mhd/sawtooth/sawtooth_models.py
torax/_src/mhd/sawtooth/sawtooth_solver.py
torax/_src/mhd/sawtooth/simple_redistribution.py
torax/_src/mhd/sawtooth/simple_trigger.py
torax/_src/mhd/sawtooth/trigger_base.py
torax/_src/neoclassical/config.py
torax/_src/neoclassical/neoclassical_intermediates.py
torax/_src/neoclassical/neoclassical_model.py
torax/_src/neoclassical/register_model.py
torax/_src/neoclassical/runtime_params.py
torax/_src/neoclassical/bootstrap_current/__init__.py
torax/_src/neoclassical/bootstrap_current/base.py
torax/_src/neoclassical/bootstrap_current/redl.py
torax/_src/neoclassical/bootstrap_current/runtime_params.py
torax/_src/neoclassical/bootstrap_current/sauter.py
torax/_src/neoclassical/bootstrap_current/zeros.py
torax/_src/neoclassical/conductivity/base.py
torax/_src/neoclassical/conductivity/redl.py
torax/_src/neoclassical/conductivity/runtime_params.py
torax/_src/neoclassical/conductivity/sauter.py
torax/_src/neoclassical/formulas/__init__.py
torax/_src/neoclassical/formulas/formulas.py
torax/_src/neoclassical/formulas/redl.py
torax/_src/neoclassical/formulas/sauter.py
torax/_src/neoclassical/poloidal_velocity/__init__.py
torax/_src/neoclassical/poloidal_velocity/base.py
torax/_src/neoclassical/poloidal_velocity/kim.py
torax/_src/neoclassical/poloidal_velocity/runtime_params.py
torax/_src/neoclassical/poloidal_velocity/zeros.py
torax/_src/neoclassical/transport/__init__.py
torax/_src/neoclassical/transport/angioni_sauter.py
torax/_src/neoclassical/transport/base.py
torax/_src/neoclassical/transport/runtime_params.py
torax/_src/neoclassical/transport/zeros.py
torax/_src/orchestration/adaptive_step.py
torax/_src/orchestration/explicit_model_outputs.py
torax/_src/orchestration/initial_state.py
torax/_src/orchestration/jit_run_loop.py
torax/_src/orchestration/run_simulation.py
torax/_src/orchestration/sawtooth_step.py
torax/_src/orchestration/sim_error.py
torax/_src/orchestration/sim_state.py
torax/_src/orchestration/state_evaluation.py
torax/_src/orchestration/step_function.py
torax/_src/orchestration/step_function_processing.py
torax/_src/orchestration/whilei_loop.py
torax/_src/output_tools/__init__.py
torax/_src/output_tools/impurity_radiation.py
torax/_src/output_tools/output.py
torax/_src/output_tools/output_grid_context.py
torax/_src/output_tools/output_keys.py
torax/_src/output_tools/post_processing.py
torax/_src/output_tools/safety_factor_fit.py
torax/_src/pedestal_model/config.py
torax/_src/pedestal_model/pedestal_model.py
torax/_src/pedestal_model/pedestal_model_output.py
torax/_src/pedestal_model/pedestal_state.py
torax/_src/pedestal_model/register_model.py
torax/_src/pedestal_model/runtime_params.py
torax/_src/pedestal_model/formation/__init__.py
torax/_src/pedestal_model/formation/base.py
torax/_src/pedestal_model/formation/power_scaling_formation_model.py
torax/_src/pedestal_model/formation/prescribed_formation_model.py
torax/_src/pedestal_model/saturation/__init__.py
torax/_src/pedestal_model/saturation/base.py
torax/_src/pedestal_model/saturation/profile_value_saturation_model.py
torax/_src/pedestal_model/structure/__init__.py
torax/_src/pedestal_model/structure/base.py
torax/_src/pedestal_model/structure/no_pedestal.py
torax/_src/pedestal_model/structure/pedestal_structure.py
torax/_src/pedestal_model/structure/set_ptop_ttopratio_ntop.py
torax/_src/pedestal_model/structure/set_ttop_ntop.py
torax/_src/physics/__init__.py
torax/_src/physics/charge_states.py
torax/_src/physics/collisions.py
torax/_src/physics/fast_ion.py
torax/_src/physics/formulas.py
torax/_src/physics/psi_calculations.py
torax/_src/physics/rotation.py
torax/_src/physics/scaling_laws.py
torax/_src/physics/radiation/__init__.py
torax/_src/physics/radiation/mavrin_coronal_cooling_rate.py
torax/_src/physics/radiation/mavrin_noncoronal_cooling_rate.py
torax/_src/physics/radiation/radiation.py
torax/_src/plotting/plotruns_lib.py
torax/_src/plotting/plotruns_lib_test.py
torax/_src/solver/__init__.py
torax/_src/solver/config.py
torax/_src/solver/fixed_point_method.py
torax/_src/solver/jax_fixed_point.py
torax/_src/solver/jax_root_finding.py
torax/_src/solver/linear_theta_method.py
torax/_src/solver/linesearch.py
torax/_src/solver/nonlinear_theta_method.py
torax/_src/solver/runtime_params.py
torax/_src/solver/solver.py
torax/_src/solver/solver_numeric_outputs.py
torax/_src/sources/__init__.py
torax/_src/sources/base.py
torax/_src/sources/bremsstrahlung_heat_sink.py
torax/_src/sources/config.py
torax/_src/sources/core_sources.py
torax/_src/sources/cyclotron_radiation_heat_sink.py
torax/_src/sources/electron_cyclotron_source.py
torax/_src/sources/formulas.py
torax/_src/sources/fusion_heat_source.py
torax/_src/sources/gas_puff_source.py
torax/_src/sources/generic_current_source.py
torax/_src/sources/generic_ion_el_heat_source.py
torax/_src/sources/generic_particle_source.py
torax/_src/sources/neutral_beam_injection_source.py
torax/_src/sources/ohmic_heat_source.py
torax/_src/sources/particle_feedback_source.py
torax/_src/sources/pellet_source.py
torax/_src/sources/qei_source.py
torax/_src/sources/register_model.py
torax/_src/sources/runtime_params.py
torax/_src/sources/source.py
torax/_src/sources/source_models.py
torax/_src/sources/source_profile_builders.py
torax/_src/sources/source_state.py
torax/_src/sources/impurity_radiation_heat_sink/__init__.py
torax/_src/sources/impurity_radiation_heat_sink/impurity_radiation_constant_fraction.py
torax/_src/sources/impurity_radiation_heat_sink/impurity_radiation_heat_sink.py
torax/_src/sources/impurity_radiation_heat_sink/impurity_radiation_mavrin_fit.py
torax/_src/sources/ion_cyclotron_source/__init__.py
torax/_src/sources/ion_cyclotron_source/base.py
torax/_src/sources/ion_cyclotron_source/scaled_profile.py
torax/_src/test_utils/core_profile_helpers.py
torax/_src/test_utils/default_configs.py
torax/_src/test_utils/default_sources.py
torax/_src/test_utils/paths.py
torax/_src/test_utils/sim_test_case.py
torax/_src/test_utils/source_helpers.py
torax/_src/test_utils/torax_refs.py
torax/_src/time_step_calculator/__init__.py
torax/_src/time_step_calculator/chi_time_step_calculator.py
torax/_src/time_step_calculator/config.py
torax/_src/time_step_calculator/fixed_time_step_calculator.py
torax/_src/time_step_calculator/from_previous_time_step_calculator.py
torax/_src/time_step_calculator/runtime_params.py
torax/_src/time_step_calculator/time_step_calculator.py
torax/_src/time_step_calculator/time_step_calculator_state.py
torax/_src/torax_pydantic/__init__.py
torax/_src/torax_pydantic/config.py
torax/_src/torax_pydantic/config_base.py
torax/_src/torax_pydantic/file_restart.py
torax/_src/torax_pydantic/interpolated_param_1d.py
torax/_src/torax_pydantic/interpolated_param_2d.py
torax/_src/torax_pydantic/pydantic_types.py
torax/_src/torax_pydantic/torax_pydantic.py
torax/_src/transport_model/__init__.py
torax/_src/transport_model/base_qlknn_model.py
torax/_src/transport_model/bohm_gyrobohm.py
torax/_src/transport_model/component.py
torax/_src/transport_model/config.py
torax/_src/transport_model/config_base.py
torax/_src/transport_model/critical_gradient.py
torax/_src/transport_model/enums.py
torax/_src/transport_model/pereverzev.py
torax/_src/transport_model/prescribed.py
torax/_src/transport_model/qlknn_10d.py
torax/_src/transport_model/qlknn_model_wrapper.py
torax/_src/transport_model/qlknn_transport_model.py
torax/_src/transport_model/qualikiz_based_transport_model.py
torax/_src/transport_model/qualikiz_transport_model.py
torax/_src/transport_model/quasilinear_utils.py
torax/_src/transport_model/register_model.py
torax/_src/transport_model/runtime_params.py
torax/_src/transport_model/tglf_based_transport_model.py
torax/_src/transport_model/tglfnn_ukaea_transport_model.py
torax/_src/transport_model/transport_coefficients_builder.py
torax/_src/transport_model/transport_coeffs.py
torax/_src/transport_model/transport_model.py
torax/_src/transport_model/transport_state.py
torax/_src/transport_model/tglf/__init__.py
torax/_src/transport_model/tglf/defaults.py
torax/_src/transport_model/tglf/tglf2py.py
torax/_src/transport_model/tglf/tglf2py_lib.pyi
torax/_src/transport_model/tglf/tglf_transport_model.py
torax/benchmarks/iterhybrid_predictor_corrector_fixed_small_dt.py
torax/benchmarks/tokagrad_benchmark.py
torax/data/third_party/geo/README.md
torax/data/third_party/geo/STEP_SPP_001_ECHD_ftop.eqdsk
torax/data/third_party/geo/iterhybrid.mat2cols
torax/data/third_party/geo/iterhybrid_cocos02.eqdsk
torax/data/third_party/geo/iterhybrid_cocos11.eqdsk
torax/data/third_party/geo/iterhybrid_cocos17.eqdsk
torax/data/third_party/imas/ITERhybrid_COCOS17_IDS_ddv4.nc
torax/data/third_party/imas/ITERhybrid_rampup_11_time_slices_COCOS17_IDS_ddv4.nc
torax/data/third_party/imas/README
torax/data/third_party/imas/STEP_SPP_001_ECHD_ftop.nc
torax/data/third_party/imas/core_profiles_15MA_DT_50_50_flat_top_slice.nc
torax/data/third_party/imas/core_profiles_ddv4_iterhybrid_rampup_conditions.nc
torax/examples/basic_config.py
torax/examples/iterhybrid_predictor_corrector.py
torax/examples/iterhybrid_rampup.py
torax/examples/step_flattop_bgb.py
torax/experimental/__init__.py
torax/experimental/edge.py
torax/experimental/fast_ion.py
torax/experimental/gas_puff_feedback_source.py
torax/experimental/geometry.py
torax/experimental/jax_utils.py
torax/experimental/neoclassical.py
torax/plotting/__init__.py
torax/plotting/plotruns.py
torax/plotting/configs/default_plot_config.py
torax/plotting/configs/global_params_plot_config.py
torax/plotting/configs/simple_plot_config.py
torax/plotting/configs/sources_plot_config.py
torax/plotting/configs/transport_plot_config.py