Repository navigation

All pull requests

Introduce ExplicitModelOutputs to bundle explicit source and pedestal outputs.
#2737copybara-service[bot] opened Oct 21, 2026Bot
Add FormationState and PowerScalingFormationState and compute smoothed thermal energy derivative in PowerScalingFormationModel.
#2736copybara-service[bot] opened Oct 21, 2026Bot
Remove internal_plasma_energy from CoreProfiles and compute thermal energy outputs in post_processing.py.
#2735copybara-service[bot] opened Oct 20, 2026Bot
Enforce consistent arithmetic using fixed k-loop sizes. Reduction dimensions are now only dependent on input shape and data type, and not CPU-dependent parameters like kernel block sizes.
#2733copybara-service[bot] opened Oct 20, 2026Bot
Remove unused prev_core_profiles (core_profiles_t) and dt parameters from updaters.update_core_profiles_during_step, CoeffsCallback.__call__, predictor_corrector_method, theta_method_block_residual, theta_method_block_loss, and jaxopt_solver.
#2732copybara-service[bot] opened Oct 20, 2026Bot
Update PedestalModel and callers to use PedestalState.
#2731copybara-service[bot] opened Oct 19, 2026Bot
Refactor StateHistory.simulation_output_to_xr helpers in output.py
#2730copybara-service[bot] opened Oct 19, 2026Bot
Simplify test setup in fvm_test.py using initial_state._get_initial_state.
#2728copybara-service[bot] opened Oct 17, 2026Bot
Add core_profile_helpers.make_initial_inputs and use it in unit tests.
#2727copybara-service[bot] opened Oct 17, 2026Bot
Use a heuristic (if possible) to determine the max steps.
#2724copybara-service[bot] opened Oct 16, 2026Bot
Rename j_* toroidal current density variables and outputs to j_toroidal_*
#2717copybara-service[bot] opened Oct 15, 2026Bot
Rename ne_tau to n_e_tau across extended Lengyel and radiation models.
#2716copybara-service[bot] opened Oct 15, 2026Bot
Pass source_state into Source._get_model_value.
#2692copybara-service[bot] opened Oct 15, 2026Bot
Update TransportModel and transport coefficients builder with TransportState.
#2691copybara-service[bot] opened Oct 15, 2026Bot
Thread component_state into ComponentTransportModel.call_implementation and all transport model implementations so they receive and return ComponentTransportState.
#2690copybara-service[bot] opened Oct 15, 2026Bot
Make source_states required in initial_core_profiles.
#2689copybara-service[bot] opened Oct 15, 2026Bot
Update Source.get_value and source profile builders with SourceState.
#2678copybara-service[bot] opened Oct 15, 2026Bot
Thread ModelState through FVM and Solvers.
#2677copybara-service[bot] opened Oct 15, 2026Bot
Thread ModelState through Orchestration and Outputs.
#2676copybara-service[bot] opened Oct 15, 2026Bot
Add initial_state methods.
#2675copybara-service[bot] opened Oct 15, 2026Bot
Add PedestalState and ModelState dataclasses.
#2639copybara-service[bot] opened Oct 12, 2026Bot
Add SourceState and TransportState dataclasses.
#2638copybara-service[bot] opened Oct 12, 2026Bot
Allow explicit IMAS core source identifier overrides
#2602sylvesterkaczmarek opened Oct 9, 2026
Support explicit IMAS ion aliases when importing plasma composition
#2601sylvesterkaczmarek opened Oct 9, 2026
Validate mandatory IMAS equilibrium geometry fields before calculations
#2600sylvesterkaczmarek opened Oct 9, 2026

Navigation Menu

Repository navigation

All pull requests

Introduce `ExplicitModelOutputs` to bundle explicit source and pedestal outputs.

Add `FormationState` and `PowerScalingFormationState` and compute smoothed thermal energy derivative in `PowerScalingFormationModel`.

Remove `internal_plasma_energy` from `CoreProfiles` and compute thermal energy outputs in `post_processing.py`.

Enforce consistent arithmetic using fixed k-loop sizes. Reduction dimensions are now only dependent on input shape and data type, and not CPU-dependent parameters like kernel block sizes.

Remove unused `prev_core_profiles` (`core_profiles_t`) and `dt` parameters from `updaters.update_core_profiles_during_step`, `CoeffsCallback.call`, `predictor_corrector_method`, `theta_method_block_residual`, `theta_method_block_loss`, and `jaxopt_solver`.

Update PedestalModel and callers to use PedestalState.

Refactor StateHistory.simulation_output_to_xr helpers in output.py

Simplify test setup in `fvm_test.py` using `initial_state._get_initial_state`.

Add `core_profile_helpers.make_initial_inputs` and use it in unit tests.

Use a heuristic (if possible) to determine the max steps.

Rename `j_` toroidal current density variables and outputs to `j_toroidal_`

Rename ne_tau to n_e_tau across extended Lengyel and radiation models.

Pass source_state into Source._get_model_value.

Update TransportModel and transport coefficients builder with TransportState.

Thread `component_state` into `ComponentTransportModel.call_implementation` and all transport model implementations so they receive and return `ComponentTransportState`.

Make `source_states` required in `initial_core_profiles`.

Update Source.get_value and source profile builders with SourceState.

Thread ModelState through FVM and Solvers.

Thread ModelState through Orchestration and Outputs.

Add initial_state methods.

Add PedestalState and ModelState dataclasses.

Add SourceState and TransportState dataclasses.

Allow explicit IMAS core source identifier overrides

Support explicit IMAS ion aliases when importing plasma composition

Validate mandatory IMAS equilibrium geometry fields before calculations

Pull requests