xla
[SPMD] Introduce high level manual sharding APIs
#6931

Merged

[SPMD] Introduce high level manual sharding APIs #6931

alanwaketan merged 5 commits into master from alanwaketan/manual_sharding_api

Add an e2e test

0908fb31

initial commit

ce606eff

Fix linters

0bf9554e

Fix linters

651415a9

alanwaketan requested a review from

yeounoh 1 year ago

alanwaketan requested a review from

jonb377 1 year ago

alanwaketan assigned

alanwaketan 1 year ago

yeounoh commented on 2024-04-17

yeounoh approved these changes on 2024-04-17

jonb377 approved these changes on 2024-04-17

jonb377 commented on 2024-04-17

Fix comment

4aafe14a

alanwaketan merged 9b2ac4b1 into master 1 year ago

alanwaketan deleted the alanwaketan/manual_sharding_api branch 1 year ago

Reviewers

jonb377

yeounoh

Assignees

alanwaketan

Labels

None yet

Milestone

No milestone