Skip to content

Pull requests: google/torchax

New pull request New
4 Open 69 Closed
4 Open 69 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

Thread PRNG key through wrap_jax_jit to prevent UnexpectedTracerError
#95 opened May 12, 2026 by ifding Loading…
1
3
add rng option to jittablemodule
#85 opened Mar 31, 2026 by qihqi Collaborator Loading…
12
Add additional argument to _aten_bmm
#79 opened Feb 22, 2026 by jrplatin Loading…
4
add make file
#50 opened Nov 18, 2025 by ShitongWang Collaborator • Draft
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.