google-deepmind/torax

View on GitHub

TORAX: Tokamak transport simulation in JAX

Stars: 721
Forks: 145
Open beginner issues: 0
Indexed issues: 31
Avg merge: 2d 14h
Merged PRs (30d): 49

Dominant language: Python
License: No license data
Last GitHub push: Sep 11, 2026
Latest indexed: Sep 16, 2026
Contributing guide: Contributing guide
Code of conduct: No code of conduct
Beginner labels: No beginner labels indexed

Open All

31 open issues indexed

Make TORAX consistent with a specific COCOS Open

Domain expertise required

google-deepmind/torax#528 · 6 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Multiple ion transport instead of electron transport Open

Domain expertise required

google-deepmind/torax#531 · 1 comment · 0 reactions · 1 assignee · Indexed Sep 16, 2026
Implement gas puff feedback (or new state equation) for desired reference <n> or nbar Open

Domain expertise required

google-deepmind/torax#532 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Implement transport coefficient feedback (or new state equation) for desired H-factor Open

Domain expertise required

google-deepmind/torax#533 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Multi-Gaussian ECCD extension. Open

google-deepmind/torax#534 · 3 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Enable default configs for different tokamak configurations Open

Domain expertise required

google-deepmind/torax#544 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Provide helpful warnings when an unphysical config has been launched Open

google-deepmind/torax#844 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Add full "circuit equation" boundary condition on the psi equation Open

Domain expertise required

google-deepmind/torax#1611 · 3 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Smoothing functionality applied to specific transport models Open

Domain expertise required

google-deepmind/torax#1731 · 3 comments · 1 reaction · 0 assignees · Indexed Sep 16, 2026
Loading IMAS geometry from equilibrium ids object is incompatible with saving config to output Open

google-deepmind/torax#1744 · 8 comments · 0 reactions · 1 assignee · Indexed Sep 16, 2026
Populate remainder of IMAS core profiles output Open

google-deepmind/torax#1814 · 1 comment · 0 reactions · 1 assignee · Indexed Sep 16, 2026
Design discussion: Simulation checkpoint & restart support for TORAX Open

google-deepmind/torax#1894 · 6 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Mapping output structure of core_sources IDS to TORAX sources Open

Domain expertise required

google-deepmind/torax#1897 · 8 comments · 0 reactions · 1 assignee · Indexed Sep 16, 2026
TGLFNN/rotation compatibility Open

google-deepmind/torax#1939 · 24 comments · 0 reactions · 1 assignee · Indexed Sep 16, 2026
Add the collisionallity nu_star to the outputs Open

google-deepmind/torax#2094 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Handle multi-ions in IMAS mapping Open

google-deepmind/torax#2212 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
More robust handling of input IDS Open

google-deepmind/torax#2213 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Add in source identifier to IMAS output Open

google-deepmind/torax#2233 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Adapt the reading of the different ion particles sources when reading `core_sources` from IMAS to parse the different ion particle sources when ion particle transport will be implemented. Open

google-deepmind/torax#2234 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Modify the mapping in `imas_tools/output/core_profiles.py` as currently extrapolated quantities should have consistent boundary values by then. Open

google-deepmind/torax#2235 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Output `core_sources.py` to be modified to output information on the individual ion sources. Open

google-deepmind/torax#2236 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Add electron ion collision time to TORAX outputs. Open

google-deepmind/torax#2237 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Better handling of non-standard ions when loading from IMAS Open

google-deepmind/torax#2238 · 4 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Failure to map sources when loading IMAS core_sources Open

google-deepmind/torax#2239 · 4 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Extended Lengyel + core linear solver JAX issues Open

google-deepmind/torax#2270 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Live plotting Open

google-deepmind/torax#2282 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Speed improvement: pre-interpolate geometries when using `fixed_time_step` Open

google-deepmind/torax#2315 · 1 comment · 1 reaction · 0 assignees · Indexed Sep 16, 2026
`jax.jacfwd` and `jax.jacrev(jax.grad(...))` return NaN through `experimental.run_loop_jit`, while `jax.grad` works fine Open

google-deepmind/torax#2331 · 5 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Suggestion: a structured docs site built from your README + readthedocs content Open

google-deepmind/torax#2349 · 0 comments · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Make geometry builders differentiable Open

google-deepmind/torax#2356 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 16, 2026
Check that required fields are present in IMAS geometry loader Open

google-deepmind/torax#2409 · 1 comment · 0 reactions · 0 assignees · Indexed Sep 16, 2026

google-deepmind/torax

Get new issues in your inbox