Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
5 Open 55 Closed
5 Open 55 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 nd_loop and Enable block_n tiling for all_gather_lhs_matmul kokoro:force-run pull ready Ready for copybara import and testing
#29822 opened Jun 27, 2025 by hanzlfs Loading… updated Aug 7, 2025
6
Retry connection errors to combat flakiness in google metadata server during tpu initialization (reopens #24432)
#26919 opened Mar 4, 2025 by dlwh Loading… updated Mar 7, 2025
2
Refactor & Document
#21691 opened Jun 6, 2024 by Arkay92 Loading… updated Jun 6, 2024
2
Update JAX README - add shard_map information (pmap deprecation)
#17210 opened Aug 21, 2023 by 8bitmp3 Loading… updated Aug 22, 2023
2
Extend Common-Gotchas docs: speed of jnp.array
#11054 opened Jun 10, 2022 by onnoeberhard Loading… updated Oct 4, 2022
1
4
ProTip! no:milestone will show everything without a milestone.