PR #7044 [FSDPv2] Support MultiSlice

[FSDPv2] Support MultiSlice #7044

alanwaketan merged 4 commits into master from alanwaketan/fsdp2_ms

alanwaketan requested a review from

jonb377 1 year ago

alanwaketan assigned

alanwaketan 1 year ago

initial commit

5f8063af

Fix linters

aeb59b5d

Add fsdpv2 test to tpu ci

76c7435b

alanwaketan force pushed from 088da28c to 76c7435b 1 year ago

JackCaoG added tpuci

jonb377 approved these changes on 2024-05-10

Fix comment

e1a5d760

alanwaketan merged 6f0b61e5 into master 1 year ago

Reviewers

jonb377

Assignees

alanwaketan

Labels

None yet

Milestone

No milestone