-
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
Rename transport_model_params to core_transport_model_params in CombinedRuntimeParams.
#2361
opened Aug 12, 2026 by
copybara-service
Bot
Loading…
Migrate all imports from transport_model to component.
#2360
opened Aug 12, 2026 by
copybara-service
Bot
Loading…
Add __call__ and encapsulate CHANNEL_CONFIG in ComponentTransportModel.
#2359
opened Aug 12, 2026 by
copybara-service
Bot
Loading…
Replace temporary xla_metadata calls with
jax.jit(f, inline=jax.Inline.XLA_LATE).
#2357
opened Aug 12, 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' be differentiable through closures.
#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…
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…
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…
Add mapping for output core_transport to IMAS
#2224
opened Jun 18, 2026 by
MateoBell
Collaborator
Loading…
(PR 4/4): Implement IMAS output core_sources mapping
#2222
opened Jun 17, 2026 by
MateoBell
Collaborator
Loading…
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 explicit data dictionary version dependency.
#2155
opened May 6, 2026 by
copybara-service
Bot
Loading…
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…
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…
Previous Next
ProTip!
Updated in the last three days: updated:>2026-08-10.