-
Notifications
You must be signed in to change notification settings - Fork 40
Pull requests: google-deepmind/torax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Make TimeVaryingArray and TimeVaryingScalar immutable. This is with a view to make the full model config immutable, which makes it possible to safely cache computations inside the model without worrying that a user has modified the field of eg. TimeVaryingArray, which is undetectable from a higher-level without major overhead.
#730
opened Feb 12, 2025 by
copybara-service
bot
Loading…
Add rhonorm1_defined_in_timerhoinput method to TimeVaryingArray.
#727
opened Feb 10, 2025 by
copybara-service
bot
Loading…
Optimize array stacking in the output by using NumPy rather than JAX.
#586
opened Dec 2, 2024 by
copybara-service
bot
Loading…
[WIP] Add TGLF transport models and surrogates
#477
opened Oct 30, 2024 by
theo-brown
•
Draft
2 of 4 tasks
ProTip!
What’s not been updated in a month: updated:<2025-01-22.