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

[mutable-arrays] persist shardings through xla computations #26377

Merged

Conversation

mattjj
Copy link
Collaborator

@mattjj mattjj commented Feb 7, 2025

Fixes #26338
Fixes #26350
Fixes #26408

@mattjj mattjj requested a review from dougalm February 7, 2025 00:05
@mattjj mattjj force-pushed the maintain-mutable-array-sharding branch from 4b5cd99 to cc94574 Compare February 7, 2025 00:08
@mattjj mattjj added the pull ready Ready for copybara import and testing label Feb 7, 2025
@mattjj mattjj force-pushed the maintain-mutable-array-sharding branch from cc94574 to 719031c Compare February 7, 2025 18:33
@copybara-service copybara-service bot merged commit c0ba362 into jax-ml:main Feb 7, 2025
18 of 19 checks passed
@mattjj mattjj deleted the maintain-mutable-array-sharding branch February 8, 2025 14:27
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
1 participant