Divide SimulationStepFn call into phases and make them public. This will allow external callers to be able to inject/modify dynamic runtime params slices at each step.

google-deepmind / torax

TORAX: Tokamak transport simulation in JAX

https://torax.readthedocs.io

Other

345 stars 31 forks source link

Divide SimulationStepFn call into phases and make them public. This will allow external callers to be able to inject/modify dynamic runtime params slices at each step. #224

Closed copybara-service[bot] closed 3 months ago

copybara-service[bot] commented 3 months ago

Divide SimulationStepFn call into phases and make them public. This will allow external callers to be able to inject/modify dynamic runtime params slices at each step.

We also make merge_source_profiles and get_initial_source_profiles public.