xla
[FSDPv2] Enable auto-wrapping
#6499
Merged

[FSDPv2] Enable auto-wrapping #6499

alanwaketan merged 3 commits into master from alanwaketan/fsdp2_auto_wrap
alanwaketan
alanwaketan [FSDPv2] Enable auto-wrapping
7ccbf8f8
alanwaketan alanwaketan requested a review from yeounoh yeounoh 1 year ago
alanwaketan alanwaketan requested a review from jonb377 jonb377 1 year ago
alanwaketan alanwaketan assigned alanwaketan alanwaketan 1 year ago
alanwaketan Fix the tests
b4eb6c1a
alanwaketan Fix the linters
114d17ef
alanwaketan alanwaketan marked this pull request as ready for review 1 year ago
jonb377
jonb377 approved these changes on 2024-02-08
alanwaketan
alanwaketan alanwaketan merged cbe0ca84 into master 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
Milestone