Skip to content

Pull requests: google-deepmind/torax

New pull request New
26 Open 1,475 Closed
26 Open 1,475 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

Fix bugs in Solver hash and eq
#1634 opened Oct 8, 2025 by copybara-service bot Loading…
fix bug where dataclasses overwrites __hash__
#1633 opened Oct 8, 2025 by copybara-service bot Loading…
Fix bug where dataclasses decorator overwrote custom hash
#1632 opened Oct 8, 2025 by copybara-service bot Loading…
Validator for profile_conditions
#1630 opened Oct 8, 2025 by MateoBell • Draft
1 of 5 tasks
1
5
Add design doc explaining how to make classes for use with jax.jit
#1629 opened Oct 7, 2025 by copybara-service bot Loading…
Add documentation on JIT compatible objects in TORAX.
#1627 opened Oct 7, 2025 by copybara-service bot Loading…
Make PedestalModel a frozen dataclass + pytree
#1626 opened Oct 6, 2025 by copybara-service bot Loading…
example sim test for imas core_profiles
#1621 opened Oct 6, 2025 by MateoBell • Draft
Enable use_jax_custom_root by default.
#1620 opened Oct 6, 2025 by copybara-service bot Loading…
plasma_composition converter for imas core_profiles
#1619 opened Oct 6, 2025 by MateoBell Loading…
7
Move extended_lengyel solver methods to new module.
#1618 opened Oct 6, 2025 by copybara-service bot Loading…
Refactor DivertorSOL1D into separate Parameters and State dataclasses.
#1617 opened Oct 6, 2025 by copybara-service bot Loading…
Move adaptive step to use whilei_loop.
#1609 opened Oct 2, 2025 by copybara-service bot Loading…
Introduce PedestalPolicy.
#1583 opened Sep 23, 2025 by copybara-service bot Loading…
Add COCOS conversion for EQDSK copybara:import-manual Set when ready for copybara manual import
#1506 opened Sep 4, 2025 by theo-brown Loading…
20
Switch psi sources to being <j.B>/B_0
#1489 opened Sep 1, 2025 by theo-brown Loading…
6 of 15 tasks
2
14
Stricter typing for geometry.
#1479 opened Aug 28, 2025 by copybara-service bot Loading…
Merge fixed step and adaptive step logic and remove fixed step.
#1464 opened Aug 22, 2025 by copybara-service bot Loading…
Add a BatchedCellVariable class.
#1444 opened Aug 14, 2025 by copybara-service bot Loading…
Implementation and testing for reading and writing IMAS core/plasma_profiles IDS
#1443 opened Aug 14, 2025 by MateoBell • Draft
7 of 9 tasks
45
Feature/muscle3 actor
#1439 opened Aug 13, 2025 by mikesndrs Loading…
9
Add TGLFNNukaeaTransportModel
#1436 opened Aug 12, 2025 by theo-brown Loading…
15
Allow plotting with only a single time point
#1376 opened Jul 21, 2025 by Karel-van-de-Plassche Loading…
1
5
feat(plotting): Enable interactive time slider in notebooks using Bokeh
#1314 opened Jun 22, 2025 by Akshay-Akz2004 Loading…
3 tasks done
1
1
Move testdata to _src
#1226 opened May 19, 2025 by copybara-service bot Loading…
ProTip! Follow long discussions with comments:>50.