Skip to content

Pull requests: jax-ml/jax

New pull request New
Clear current search query, filters, and sorts
1 Open 75 Closed
1 Open 75 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

all_to_all: Add optional axis_index_groups argument cla: yes pull ready Ready for copybara import and testing
#5059 by jpuigcerver was merged Dec 7, 2020 Loading…
@apaszke
16
Support mapping a single logical axis to multiple mesh axes cla: yes pull ready Ready for copybara import and testing
#5099 by apaszke was merged Dec 11, 2020 Loading…
@apaszke
2
Clean up xmap tests cla: yes pull ready Ready for copybara import and testing
#5435 by apaszke was merged Jan 18, 2021 Loading…
@apaszke
1
Make it possible to vmap xmapped functions cla: yes pull ready Ready for copybara import and testing
#5453 by apaszke was merged Jan 19, 2021 Loading…
@apaszke
2
Restore support for nested xmaps when using the (standard) replica lowering cla: yes pull ready Ready for copybara import and testing
#5465 by apaszke was merged Jan 25, 2021 Loading…