Skip to content

Pull requests: google-deepmind/torax

New pull request New
49 Open 2,047 Closed
49 Open 2,047 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add units to TORAX outputs.
#2278 opened Jul 8, 2026 by copybara-service Bot Loading…
Add output key constants for edge model scalars and main_ion_fractions
#2277 opened Jul 8, 2026 by copybara-service Bot Loading…
Refactor: extract output DataTree keys to a dedicated output_keys module
#2276 opened Jul 8, 2026 by copybara-service Bot Loading…
Add collisionality_max to Qualikiz and TGLF based transport models.
#2275 opened Jul 7, 2026 by copybara-service Bot Loading…
Adding type suppressions for pyrefly
#2274 opened Jul 7, 2026 by copybara-service Bot Loading…
Bounded while_loop implementation using jax.lax.while_loop and a custom_vjp.
#2273 opened Jul 7, 2026 by copybara-service Bot Loading…
Migrate ITER Predictor-Corrector configs to combined transport model.
#2260 opened Jun 29, 2026 by copybara-service Bot Loading…
Add tau_ei to TORAX outputs
#2259 opened Jun 29, 2026 by dylantirandaz Loading…
1
Add nu_star to TORAX outputs
#2258 opened Jun 29, 2026 by dylantirandaz Loading…
1
Migrate default and simple configs to combined transport model.
#2256 opened Jun 29, 2026 by copybara-service Bot Loading…
Migrate SPARC configs to combined transport model.
#2254 opened Jun 29, 2026 by copybara-service Bot Loading…
Migrate ITER configs to combined transport model.
#2253 opened Jun 29, 2026 by copybara-service Bot Loading…
Code update
#2252 opened Jun 29, 2026 by copybara-service Bot Loading…
feat: add q-profile warnings for unphysical configs (Issue #844)
#2243 opened Jun 23, 2026 by Paramveersingh-S Loading…
1
fix: support plotting initial conditions and empty datasets (Issue #1…
#2242 opened Jun 23, 2026 by Paramveersingh-S Loading…
1
Followup PR: Implement IMAS output core_sources mapping
#2231 opened Jun 18, 2026 by MateoBell Collaborator Loading…
(PR 2/2) Add main_ions contribution to output core_profiles mapping
#2227 opened Jun 18, 2026 by MateoBell Collaborator Loading…
1
Add mapping for output core_transport to IMAS
#2224 opened Jun 18, 2026 by MateoBell Collaborator Loading…
6
(PR 4/4): Implement IMAS output core_sources mapping
#2222 opened Jun 17, 2026 by MateoBell Collaborator Loading…
docs: add Windows troubleshooting guide and NetCDF group usage
#2210 opened Jun 14, 2026 by chenghsiang68-netizen Loading…
3
Add normalized collisionality nu_star to outputs
#2203 opened Jun 9, 2026 by r1cksync Loading…
1
fix: use left-biased 3-point stencil for face gradients on non-uniform grids
#2202 opened Jun 9, 2026 by copybara-service Bot Loading…
Add gyaradax plugin
#2189 opened Jun 3, 2026 by gerkone • Draft
3 of 7 tasks
1
Add explicit data dictionary version dependency.
#2155 opened May 6, 2026 by copybara-service Bot Loading…
2
Expose Anderson acceleration options in the config.
#2154 opened May 6, 2026 by copybara-service Bot Loading…
ProTip! no:milestone will show everything without a milestone.