Skip to content

Pull requests: google/torchax

New pull request New
5 Open 68 Closed
5 Open 68 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
2
bump version to 0.0.13.
#92 opened May 4, 2026 by wdhongtw Collaborator Loading…
add rng option to jittablemodule
#85 opened Mar 31, 2026 by qihqi Collaborator Loading…
8
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! Exclude everything labeled bug with -label:bug.