Skip to content

Pull requests: google-deepmind/torax

New pull request New
62 Open 2,146 Closed
62 Open 2,146 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

Code update
#2397 opened Aug 22, 2026 by copybara-service Bot Loading…
Update hijax primitive name
#2396 opened Aug 22, 2026 by copybara-service Bot Loading…
Update hijax primitive name
#2395 opened Aug 22, 2026 by copybara-service Bot Loading…
Rename VJPHiPrimitive -> HiPrim.
#2394 opened Aug 21, 2026 by copybara-service Bot Loading…
[No functional change] Delete VJPHiPrimitive and replace with HiPrim. This is a simple replace.
#2393 opened Aug 21, 2026 by copybara-service Bot Loading…
Rename constant transport model to prescribed across TORAX.
#2392 opened Aug 21, 2026 by copybara-service Bot Loading…
Remove QLKNNRuntimeConfigInputs and unused Ped_top, set_pedestal from QLKNN.
#2391 opened Aug 21, 2026 by copybara-service Bot Loading…
Replace chex.Numeric with jax.typing.ArrayLike
#2390 opened Aug 21, 2026 by copybara-service Bot Loading…
Use new Python 3.12 syntax for generics.
#2389 opened Aug 21, 2026 by copybara-service Bot Loading…
[No functional change] Delete VJPHiPrimitive and replace with HiPrim. This is a simple replace.
#2388 opened Aug 21, 2026 by copybara-service Bot Loading…
Add helper methods to extract two-point face masks for Pedestal and Internal Boundary Conditions.
#2387 opened Aug 20, 2026 by copybara-service Bot Loading…
Propagate two-point face masks to transport model gradient calculations.
#2386 opened Aug 20, 2026 by copybara-service Bot Loading…
Fix test failures under LAPACK 3.12.1.
#2384 opened Aug 19, 2026 by copybara-service Bot Loading…
Add TokaMaker geometry interface
#2383 opened Aug 19, 2026 by fs2819 Loading…
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…
Add two_point_mask support to CellVariable.face_grad.
#2380 opened Aug 19, 2026 by copybara-service Bot Loading…
Decentralize source and bootstrap current output serialization to xarray.
#2377 opened Aug 19, 2026 by copybara-service Bot Loading…
Add p_ei to QeiInfo and de-duplicate electron-ion exchange calculations across TORAX.
#2378 opened Aug 19, 2026 by copybara-service Bot Loading…
Implement general left face gradient constraint calculation for CellVariable.
#2375 opened Aug 17, 2026 by copybara-service Bot Loading…
Move Extended Lengyel output keys to extended_lengyel_standalone.py.
#2368 opened Aug 14, 2026 by copybara-service Bot Loading…
Decentralize EdgeModelOutputs and ExtendedLengyelOutputs to xarray conversion.
#2367 opened Aug 14, 2026 by copybara-service Bot Loading…
Add OutputGridContext and fast vectorized dataset construction.
#2366 opened Aug 14, 2026 by copybara-service Bot Loading…
Add GridType enum and OutputKey grid metadata across output_keys.py.
#2365 opened Aug 14, 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…
ProTip! Add no:assignee to see everything that’s not assigned.