LICENSE
README.md
pyproject.toml
torax/__init__.py
torax/conftest.py
torax/run_simulation_main.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/jax_utils.py
torax/_src/math_utils.py
torax/_src/simulation_app.py
torax/_src/state.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/plasma_composition.py
torax/_src/config/profile_conditions.py
torax/_src/config/runtime_params_slice.py
torax/_src/config/runtime_validation_utils.py
torax/_src/core_profiles/getters.py
torax/_src/core_profiles/initialization.py
torax/_src/core_profiles/updaters.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/circular_geometry.py
torax/_src/geometry/geometry.py
torax/_src/geometry/geometry_loader.py
torax/_src/geometry/geometry_provider.py
torax/_src/geometry/pydantic_model.py
torax/_src/geometry/standard_geometry.py
torax/_src/mhd/base.py
torax/_src/mhd/pydantic_model.py
torax/_src/mhd/runtime_params.py
torax/_src/mhd/sawtooth/flatten_profile.py
torax/_src/mhd/sawtooth/pydantic_model.py
torax/_src/mhd/sawtooth/redistribution_base.py
torax/_src/mhd/sawtooth/runtime_params.py
torax/_src/mhd/sawtooth/sawtooth_model.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/pydantic_model.py
torax/_src/neoclassical/runtime_params.py
torax/_src/neoclassical/bootstrap_current/base.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/sauter.py
torax/_src/orchestration/initial_state.py
torax/_src/orchestration/run_loop.py
torax/_src/orchestration/run_simulation.py
torax/_src/orchestration/sim_state.py
torax/_src/orchestration/step_function.py
torax/_src/output_tools/output.py
torax/_src/output_tools/post_processing.py
torax/_src/output_tools/safety_factor_fit.py
torax/_src/pedestal_model/no_pedestal.py
torax/_src/pedestal_model/pedestal_model.py
torax/_src/pedestal_model/pydantic_model.py
torax/_src/pedestal_model/runtime_params.py
torax/_src/pedestal_model/set_pped_tpedratio_nped.py
torax/_src/pedestal_model/set_tped_nped.py
torax/_src/physics/__init__.py
torax/_src/physics/charge_states.py
torax/_src/physics/collisions.py
torax/_src/physics/formulas.py
torax/_src/physics/psi_calculations.py
torax/_src/physics/scaling_laws.py
torax/_src/plotting/plotruns_lib.py
torax/_src/plotting/plotruns_lib_test.py
torax/_src/solver/__init__.py
torax/_src/solver/linear_theta_method.py
torax/_src/solver/nonlinear_theta_method.py
torax/_src/solver/predictor_corrector_method.py
torax/_src/solver/pydantic_model.py
torax/_src/solver/runtime_params.py
torax/_src/solver/solver.py
torax/_src/sources/__init__.py
torax/_src/sources/base.py
torax/_src/sources/bootstrap_current_source.py
torax/_src/sources/bremsstrahlung_heat_sink.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/ion_cyclotron_source.py
torax/_src/sources/ohmic_heat_source.py
torax/_src/sources/pellet_source.py
torax/_src/sources/pydantic_model.py
torax/_src/sources/qei_source.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_profiles.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/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/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/fixed_time_step_calculator.py
torax/_src/time_step_calculator/pydantic_model.py
torax/_src/time_step_calculator/time_step_calculator.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/model_base.py
torax/_src/torax_pydantic/model_config.py
torax/_src/torax_pydantic/pydantic_types.py
torax/_src/torax_pydantic/register_config.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/constant.py
torax/_src/transport_model/critical_gradient.py
torax/_src/transport_model/pydantic_model.py
torax/_src/transport_model/pydantic_model_base.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_transport_model.py
torax/_src/transport_model/runtime_params.py
torax/_src/transport_model/transport_model.py
torax/data/third_party/geo/EQDSK_ITERhybrid_COCOS02.eqdsk
torax/data/third_party/geo/ITER_hybrid_citrin_equil_cheasedata.mat2cols
torax/data/third_party/geo/LICENSE_PINT
torax/data/third_party/geo/README
torax/data/third_party/geo/eqdsk_cocos02.eqdsk
torax/examples/basic_config.py
torax/examples/iterhybrid_predictor_corrector.py
torax/examples/iterhybrid_rampup.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