Skip to content

Pull requests: google-deepmind/torax

New pull request New
38 Open 1,966 Closed
38 Open 1,966 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 TODO to rename predictor_corrector to Picard.
#2173 opened May 15, 2026 by copybara-service Bot Loading…
Modify IMAS output for Python 3.14 compatibility.
#2172 opened May 15, 2026 by copybara-service Bot Loading…
Add Armijo sufficient decrease criterion to jax_root_finding.py
#2170 opened May 14, 2026 by copybara-service Bot Loading…
Move the backtracking code to a separate module
#2169 opened May 14, 2026 by copybara-service Bot Loading…
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…
Add backtracking options to solver config.
#2151 opened May 6, 2026 by copybara-service Bot Loading…
Use general backtracking line search in jax_fixed_point.
#2150 opened May 6, 2026 by copybara-service Bot Loading…
Extract general backtracking line search from jax_root_finding.
#2149 opened May 6, 2026 by copybara-service Bot Loading…
Add residual norm convergence criteria to jax_fixed_point.
#2148 opened May 6, 2026 by copybara-service Bot Loading…
Expose rtol and atol in solver config.
#2147 opened May 6, 2026 by copybara-service Bot Loading…
Support dithering in ADAPTIVE_SOURCE pedestal.
#2134 opened Apr 23, 2026 by copybara-service Bot Loading…
Security: Input validation relies on assert, which can be disabled
#2132 opened Apr 22, 2026 by tomaioo Loading…
2
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
Modify IMAS integration tests to use multi time equilibrium data.
#2127 opened Apr 20, 2026 by copybara-service Bot Loading…
Compute heat sources only up to a given rho_norm in the PowerScalingFormationModel
#2105 opened Apr 7, 2026 by copybara-service Bot Loading…
Add nu_star (collisionality) to post-processed outputs
#2103 opened Apr 4, 2026 by AndrewTKent Contributor Loading…
4 tasks
1
1
Replace exponential smoothing of dW/dt with computation with a time lag.
#2098 opened Apr 2, 2026 by copybara-service Bot Loading…
Add IMAS->TORAX Datatree convertor.
#2096 opened Apr 2, 2026 by copybara-service Bot Loading…
Persist rho_norm_ped_top in PedestalTransitionState.
#2088 opened Mar 27, 2026 by copybara-service Bot Loading…
Make both right_face_value and left_face_value a property.
#2078 opened Mar 25, 2026 by copybara-service Bot Loading…
Add BallooningStabilitySaturationModel.
#2043 opened Mar 9, 2026 by copybara-service Bot Loading…
Create NoSaturationModel and NoFormationModel
#2026 opened Mar 2, 2026 by copybara-service Bot Loading…
ProTip! no:milestone will show everything without a milestone.