Skip to content

Pull requests: google-deepmind/torax

New pull request New
34 Open 1,487 Closed
34 Open 1,487 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

Add StaticDataclass.
#1655 opened Oct 16, 2025 by copybara-service bot Loading…
Add a hybrid solver mode to the Extended Lengyel model.
#1654 opened Oct 15, 2025 by copybara-service bot Loading…
Pipe solver status into Extended Lengyel outputs.
#1653 opened Oct 15, 2025 by copybara-service bot Loading…
Adds a new test case for _solve_for_qcc for unphysical state.
#1649 opened Oct 14, 2025 by copybara-service bot Loading…
Clip c_z_prefactor to zero in inverse mode lengyel and add test
#1646 opened Oct 14, 2025 by copybara-service bot Loading…
Remove last use of chex.dataclass
#1645 opened Oct 14, 2025 by copybara-service bot Loading…
Implement Newton-Raphson solver for Extended Lengyel inverse mode.
#1643 opened Oct 13, 2025 by copybara-service bot Loading…
Implement forward mode Newton-Raphson solver for Extended Lengyel model.
#1642 opened Oct 13, 2025 by copybara-service bot Loading…
Add a public API for the transport model for registering models
#1641 opened Oct 13, 2025 by copybara-service bot Loading…
First version of a step function that acts on a fixed timestep irrespective of the other timesteps that occur
#1640 opened Oct 13, 2025 by copybara-service bot Loading…
Generalize the inputs to the fixed_point to allow PyTrees.
#1639 opened Oct 13, 2025 by copybara-service bot Loading…
Add JAX implementation scipy.optimize.fixed_point.
#1638 opened Oct 13, 2025 by copybara-service bot Loading…
Add OpenSTEP example
#1636 opened Oct 10, 2025 by theo-brown Loading…
5
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
6
Add design doc explaining how to make classes for use with jax.jit
#1629 opened Oct 7, 2025 by copybara-service bot Loading…
Make PedestalModel a StaticDataclass
#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
plasma_composition converter for imas core_profiles
#1619 opened Oct 6, 2025 by MateoBell Loading…
14
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…
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…
ProTip! no:milestone will show everything without a milestone.