xla
[SPMD] Add FSDP sharding for test_train_spmd_linear_model.py
#5299

Merged

[SPMD] Add FSDP sharding for test_train_spmd_linear_model.py #5299

alanwaketan merged 1 commit into master from alanwaketan/spmd_linear

[SPMD] Add FSDP sharding for test_train_spmd_linear_model.py

89b81197

alanwaketan requested a review from

jonb377 2 years ago

alanwaketan requested a review from

JackCaoG 2 years ago

alanwaketan assigned

alanwaketan 2 years ago

JackCaoG commented on 2023-07-12

JackCaoG approved these changes on 2023-07-12

alanwaketan merged d01880c5 into master 2 years ago

Reviewers

JackCaoG

jonb377

Assignees

alanwaketan

Labels

None yet

Milestone

No milestone