Skip to content

Pull requests: google-deepmind/torax

New pull request New
43 Open 2,182 Closed
43 Open 2,182 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 JAX compatible progress bar.
#2414 opened Sep 2, 2026 by copybara-service Bot Loading…
Restructure TurbulentTransport and TransportModel output.
#2413 opened Sep 1, 2026 by copybara-service Bot Loading…
Move Extended Lengyel edge model into dedicated subfolder (CL 1).
#2412 opened Sep 1, 2026 by copybara-service Bot Loading…
1
Update ComponentTransportModel and leaf models to return TransportCoeffs.
#2408 opened Aug 28, 2026 by copybara-service Bot Loading…
Code update
#2397 opened Aug 22, 2026 by copybara-service Bot Loading…
Add TokaMaker geometry interface
#2383 opened Aug 19, 2026 by fs2819 Loading…
1
Use Python 3.12 type annotation.
#2382 opened Aug 19, 2026 by copybara-service Bot Loading…
Improve compile time
#2381 opened Aug 19, 2026 by copybara-service Bot Loading…
Split the Solver class into prepare_step and solve_step.
#2354 opened Aug 11, 2026 by copybara-service Bot Loading…
Add support for vmapped linesearch in Newton-Raphson solver.
#2353 opened Aug 11, 2026 by copybara-service Bot Loading…
Add TrappedFractionSource enum and geometry configuration field
#2325 opened Jul 31, 2026 by theo-brown Collaborator Loading…
Add a pellet-aware time step calculator. copybara:import-manual Set when ready for copybara manual import
#2313 opened Jul 28, 2026 by MateoSochard Collaborator Loading…
44
Add test for forward-mode differentiation.
#2289 opened Jul 15, 2026 by copybara-service Bot Loading…
Support different sources for trapped particle fraction copybara:import-manual Set when ready for copybara manual import
#2283 opened Jul 13, 2026 by theo-brown Collaborator Loading…
14
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
2
Add mapping for output core_transport to IMAS
#2224 opened Jun 18, 2026 by MateoBell Collaborator Loading…
9
(PR 4/4): Implement IMAS output core_sources mapping
#2222 opened Jun 17, 2026 by MateoBell Collaborator Loading…
Add normalized collisionality nu_star to outputs
#2203 opened Jun 9, 2026 by r1cksync Loading…
1
2
Add gyaradax plugin
#2189 opened Jun 3, 2026 by gerkone • Draft
3 of 7 tasks
2
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…
Add Anderson acceleration as an option in jax_fixed_point.
#2153 opened May 6, 2026 by copybara-service Bot Loading…
Add module for Anderson acceleration.
#2152 opened May 6, 2026 by copybara-service Bot Loading…
Update IMAS DD version to 4.1.1 ready for review Add to PRs when they are ready for review unsure of who to assign as reviewer
#2130 opened Apr 22, 2026 by MateoBell Collaborator Loading…
6
ProTip! Mix and match filters to narrow down what you’re looking for.