New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

sketch out all-gather primitive #3551

Closed

mattjj wants to merge 2 commits into master from all-gather-primitive

Collaborator

mattjj commented Jun 25, 2020 •

edited

Loading

We need:

either to gate this new version on jaxlib version, or else to push new jaxlib wheels and bump the minimum version;
to wait for omnistaging #3370 's improvements to axis_index to make the transpose rule work; and
to handle pytrees.


          sketch out all-gather primitive

308fc40

googlebot added the cla: yes label


          fix typo

9c4b47f

mattjj marked this pull request as draft

June 26, 2020 13:53

mattjj closed this

mattjj deleted the all-gather-primitive branch

August 23, 2021 00:37

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels