flax
[nnx] Improve spmd
#3580
Merged

[nnx] Improve spmd #3580

copybara-service merged 1 commit into main from nnx-get-named-sharding
cgarciae
cgarciae [nnx] improve spmd
3b772808
codecov-commenter
superbobry
superbobry approved these changes on 2023-12-21
chiamp
chiamp approved these changes on 2023-12-21
chiamp chiamp added pull ready
copybara-service copybara-service merged e8a14af9 into main 2 years ago
copybara-service copybara-service deleted the nnx-get-named-sharding branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone