All modules for which code is available

• torax.array_typing
• torax.config.build_runtime_params
• torax.config.config_args
• torax.config.config_loader
• torax.config.numerics
• torax.config.plasma_composition
• torax.config.profile_conditions
• torax.config.runtime_params
• torax.config.runtime_params_slice
• torax.config.runtime_validation_utils
• torax.constants
• torax.fvm.block_1d_coeffs
• torax.fvm.calc_coeffs
• torax.fvm.cell_variable
• torax.fvm.convection_terms
• torax.fvm.diffusion_terms
• torax.fvm.discrete_system
• torax.fvm.enums
• torax.fvm.fvm_conversions
• torax.fvm.implicit_solve_block
• torax.fvm.newton_raphson_solve_block
• torax.fvm.optimizer_solve_block
• torax.fvm.residual_and_loss
• torax.geometry.circular_geometry
• torax.geometry.geometry
• torax.geometry.geometry_loader
• torax.geometry.geometry_provider
• torax.geometry.pydantic_model
• torax.geometry.standard_geometry
• torax.interpolated_param
• torax.jax_utils
• torax.math_utils
• torax.orchestration.run_simulation
• torax.output
• torax.physics.charge_states
• torax.physics.collisions
• torax.physics.formulas
• torax.physics.psi_calculations
• torax.physics.scaling_laws
• torax.plotting.plotruns_lib
• torax.plotting.plotruns_lib_test
• torax.post_processing
• torax.run_simulation_main
• torax.sources.base
• torax.sources.bootstrap_current_source
• torax.sources.bremsstrahlung_heat_sink
• torax.sources.cyclotron_radiation_heat_sink
• torax.sources.electron_cyclotron_source
• torax.sources.formulas
• torax.sources.fusion_heat_source
• torax.sources.gas_puff_source
• torax.sources.generic_current_source
• torax.sources.generic_ion_el_heat_source
• torax.sources.generic_particle_source
• torax.sources.impurity_radiation_heat_sink.impurity_radiation_constant_fraction
• torax.sources.impurity_radiation_heat_sink.impurity_radiation_heat_sink
• torax.sources.impurity_radiation_heat_sink.impurity_radiation_mavrin_fit
• torax.sources.ion_cyclotron_source
• torax.sources.ohmic_heat_source
• torax.sources.pellet_source
• torax.sources.pydantic_model
• torax.sources.qei_source
• torax.sources.runtime_params
• torax.sources.source
• torax.sources.source_models
• torax.sources.source_profile_builders
• torax.sources.source_profiles
• torax.sources.tests.bootstrap_current_source_test
• torax.sources.tests.bremsstrahlung_heat_sink_test
• torax.sources.tests.constant_fraction_impurity_radiation_heat_sink_test
• torax.sources.tests.cyclotron_radiation_heat_sink_test
• torax.sources.tests.electron_cyclotron_source_test
• torax.sources.tests.fusion_heat_source_test
• torax.sources.tests.gas_puff_source_test
• torax.sources.tests.generic_current_source_test
• torax.sources.tests.generic_ion_el_heat_source_test
• torax.sources.tests.generic_particle_source_test
• torax.sources.tests.ion_cyclotron_source_test
• torax.sources.tests.mavrin_impurity_radiation_heat_sink_test
• torax.sources.tests.ohmic_heat_source_test
• torax.sources.tests.pellet_source_test
• torax.sources.tests.pydantic_model_test
• torax.sources.tests.qei_source_test
• torax.sources.tests.source_profile_builders_test
• torax.sources.tests.source_profiles_test
• torax.sources.tests.source_test
• torax.sources.tests.test_lib
• torax.state
• torax.stepper.linear_theta_method
• torax.stepper.nonlinear_theta_method
• torax.stepper.predictor_corrector_method
• torax.stepper.pydantic_model
• torax.stepper.runtime_params
• torax.stepper.stepper
• torax.stepper.tests.pydantic_model_test
• torax.time_step_calculator.chi_time_step_calculator
• torax.time_step_calculator.fixed_time_step_calculator
• torax.time_step_calculator.pydantic_model
• torax.time_step_calculator.tests.pydantic_model_test
• torax.time_step_calculator.time_step_calculator
• torax.torax_pydantic.model_config
• torax.transport_model.base_qlknn_model
• torax.transport_model.bohm_gyrobohm
• torax.transport_model.constant
• torax.transport_model.critical_gradient
• torax.transport_model.pydantic_model
• torax.transport_model.pydantic_model_base
• torax.transport_model.qlknn_10d
• torax.transport_model.qlknn_model_wrapper
• torax.transport_model.qlknn_transport_model
• torax.transport_model.qualikiz_based_transport_model
• torax.transport_model.quasilinear_transport_model
• torax.transport_model.runtime_params
• torax.transport_model.tests.bohm_gyrobohm_test
• torax.transport_model.tests.pydantic_model_test
• torax.transport_model.tests.qlknn_model_wrapper_test
• torax.transport_model.tests.qlknn_transport_model_test
• torax.transport_model.tests.qualikiz_based_transport_model_test
• torax.transport_model.tests.qualikiz_transport_model_test
• torax.transport_model.tests.quasilinear_transport_model_test
• torax.transport_model.tests.transport_model_test
• torax.transport_model.transport_model