xla
Introduce FSDPv2
#6122
Merged

Introduce FSDPv2 #6122

alanwaketan merged 5 commits into master from alanwaketan/fsdp_v2
alanwaketan
alanwaketan Add Initial PoC and test
006c48d5
alanwaketan tmp
b1d66f8b
alanwaketan Some nits
3ac7e63a
alanwaketan alanwaketan assigned alanwaketan alanwaketan 2 years ago
jonb377
jonb377 commented on 2023-12-12
yeounoh
yeounoh commented on 2023-12-12
yeounoh
yeounoh commented on 2023-12-12
yeounoh
yeounoh approved these changes on 2023-12-12
alanwaketan
JackCaoG
JackCaoG commented on 2023-12-12
JackCaoG
JackCaoG approved these changes on 2023-12-12
alanwaketan Address comments and add tests
ab44d7a9
alanwaketan Fix linters
fe6f9f21
alanwaketan alanwaketan changed the title [WIP] Introduce FSDPv2 Introduce FSDPv2 2 years ago
alanwaketan
alanwaketan alanwaketan merged 4fe9fe7e into master 2 years ago
alanwaketan

Login to write a write a comment.

Login via GitHub

Assignees
Labels
Milestone