-
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
Move top-level TransportModel implementation to transport_model.py and add combined.py shim.
#2370
opened Aug 14, 2026 by
copybara-service
Bot
Loading…
Rename CombinedTransportModel to TransportModel and CombinedRuntimeParams to RuntimeParams.
#2369
opened Aug 14, 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…
Convert transport models from sequence to mapping with user-given names.
#2364
opened Aug 14, 2026 by
copybara-service
Bot
Loading…
Update combined.py imports to use component module directly.
#2363
opened Aug 14, 2026 by
copybara-service
Bot
Loading…
Flatten __call__ and simplify _combine in CombinedTransportModel.
#2362
opened Aug 14, 2026 by
copybara-service
Bot
Loading…
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' 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…
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…
Previous Next
ProTip!
Updated in the last three days: updated:>2026-08-12.