Skip to content
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.

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

require consistent output structure in custom vmap rules #9369

Merged
merged 1 commit into from
Mar 29, 2022

Conversation

froystig
Copy link
Member

... not always a sequence.

part of #9073

@froystig froystig added the pull ready Ready for copybara import and testing label Jan 28, 2022
@froystig froystig requested a review from mattjj January 28, 2022 18:10
@froystig froystig self-assigned this Jan 28, 2022
@froystig froystig force-pushed the custom-vmap-outputs branch from 8b0282b to b2de101 Compare March 29, 2022 19:29
@copybara-service copybara-service bot merged commit 085d390 into jax-ml:main Mar 29, 2022
@froystig froystig deleted the custom-vmap-outputs branch March 29, 2022 20:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
pull ready Ready for copybara import and testing
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants