-
Notifications
You must be signed in to change notification settings - Fork 144
Pull requests: google-deepmind/torax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Replace chex.Numeric with jax.typing.ArrayLike
#2390
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…
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…
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.
#2313
opened Jul 28, 2026 by
MateoSochard
Collaborator
Loading…
Make
jax_utils.while_loop_bounded with implementation='while_loop' handle closures correctly.
#2303
opened Jul 23, 2026 by
copybara-service
Bot
Loading…
Switch
while_loop_bounded to use implementation='while_loop'.
#2295
opened Jul 20, 2026 by
copybara-service
Bot
Loading…
Add test for forward-mode differentiation.
#2289
opened Jul 15, 2026 by
copybara-service
Bot
Loading…
Previous Next
ProTip!
Exclude everything labeled
bug with -label:bug.