flax
Fix `get_partition_spec` on replicated array
#3013
Merged

Fix `get_partition_spec` on replicated array #3013

copybara-service merged 1 commit into google:main from IvyZX:pjit
IvyZX
IvyZX IvyZX requested a review from cgarciae cgarciae 2 years ago
cgarciae
cgarciae commented on 2023-04-05
cgarciae
cgarciae commented on 2023-04-05
cgarciae
cgarciae commented on 2023-04-05
IvyZX IvyZX force pushed 2 years ago
IvyZX IvyZX force pushed 2 years ago
IvyZX IvyZX force pushed 2 years ago
codecov-commenter
IvyZX IvyZX force pushed 2 years ago
IvyZX fix pspec on replicated array + add sharding utils
efe3d57c
IvyZX IvyZX force pushed to efe3d57c 2 years ago
cgarciae
cgarciae approved these changes on 2023-04-06
cgarciae cgarciae added pull ready
copybara-service copybara-service merged 332b7d04 into main 2 years ago
IvyZX IvyZX deleted the pjit branch 321 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone