Skip to content

Pull requests: google-deepmind/torax

New pull request New
17 Open 1,199 Closed
17 Open 1,199 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

Replace usage of jax.jit with jax_utils.jit.
#1313 opened Jun 21, 2025 by copybara-service bot Loading…
Refactor neoclassical models.
#1312 opened Jun 20, 2025 by copybara-service bot Loading…
Consolidate Coulomb logarithms.
#1311 opened Jun 20, 2025 by copybara-service bot Loading…
Add neoclassical models to static args of experimental compile of newton raphson and run experimental compile and sim no compile tests as intended.
#1310 opened Jun 20, 2025 by copybara-service bot Loading…
Unify adaptive step and step.
#1309 opened Jun 19, 2025 by copybara-service bot Loading…
Add experimental compile to optimizer solver.
#1307 opened Jun 19, 2025 by copybara-service bot Loading…
Add experimental compile to linear solver.
#1306 opened Jun 19, 2025 by copybara-service bot Loading…
Add neoclassical transport coefficients to output.
#1305 opened Jun 19, 2025 by copybara-service bot Loading…
Calculate neoclassical transport. Only zero model for now.
#1304 opened Jun 19, 2025 by copybara-service bot Loading…
Refactor CoreTransport
#1303 opened Jun 19, 2025 by copybara-service bot Loading…
Add experimental compile to Solver.__call__.
#1298 opened Jun 18, 2025 by copybara-service bot Loading…
Delete vscode configuration settings.
#1228 opened May 20, 2025 by copybara-service bot Loading…
6
Move testdata to _src
#1226 opened May 19, 2025 by copybara-service bot Loading…
Move integration tests to specific folder
#1225 opened May 19, 2025 by copybara-service bot Loading…
Validate wext >= 0, enforce sorted xs, and enable JAX errors by default
#1068 opened May 3, 2025 by AlankritVerma01 Loading…
1
16
Feature/imas coupling
#938 opened Apr 2, 2025 by mikesndrs Loading…
84
[WIP] Add TGLF transport models and UKAEA surrogates
#477 opened Oct 30, 2024 by theo-brown • Draft
2 of 4 tasks
27
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.