PR #20323 Add `convert_module` to FSDP

Add `convert_module` to FSDP #20323

lantiga merged 12 commits into Lightning-AI:master from tshu-w:FSDP

tshu-w requested a review from

lantiga 1 year ago

tshu-w requested a review from

Borda 1 year ago

tshu-w requested a review from

tchaton 1 year ago

tshu-w requested a review from

justusschock 1 year ago

github-actions added fabric

github-actions added pl

tshu-w force pushed from b8f1345e to 7a0c3550 1 year ago

tshu-w force pushed from 7a0c3550 to baeb5353 1 year ago

Add `convert_module` to FSDP

ab552e15

Update ChangeLog

088b9724

tshu-w force pushed from baeb5353 to 088b9724 1 year ago

lantiga added waiting on author

Merge branch 'master' into FSDP

51f2b865

lantiga requested a review from

ethanwharris 1 year ago

make plugin type check more flexible (#20186)

26d0b380

Make plugin type check more flexible (Fabric) (#20452)

4b2e2e40

Merge branch 'master' into FSDP

e97b1cdb

Merge branch 'master' into FSDP

c92fea8d

Merge branch 'master' into FSDP

d3734e6f

Merge branch 'master' into FSDP

e15ee2d6

Pin setuptools for gpu builds

a334fee5

github-actions added ci

Fix link in doc

08b695ce

github-actions added docs

Merge branch 'master' into FSDP

9cbe4739

lantiga approved these changes on 2024-12-10

lantiga merged 601c0608 into master 1 year ago

lantiga deleted the FSDP branch 1 year ago

Reviewers

lantiga

Borda

tchaton

justusschock

ethanwharris

Assignees

No one assigned

Labels

docs ci waiting on author fabric pl

Milestone

No milestone

pytorch-lightning Add `convert_module` to FSDP #20323 Merged

Add `convert_module` to FSDP #20323

pytorch-lightning
Add `convert_module` to FSDP
#20323

Merged