pytorch-lightning
feat(FSDP): enable FSDP strategy on CPU and add trainable checkpointing tests
#21840

Open

feat(FSDP): enable FSDP strategy on CPU and add trainable checkpointing tests #21840

zhixiangli wants to merge 9 commits into Lightning-AI:master from zhixiangli:feature/fsdp-cpu

zhixiangli requested a review from

tchaton 29 days ago

zhixiangli requested a review from

justusschock 29 days ago

zhixiangli requested a review from

ethanwharris 29 days ago

zhixiangli force pushed from 2aea2dcd to 05941d41 28 days ago

Allow FSDP strategy on CPU accelerator

95add982

[pre-commit.ci] auto fixes from pre-commit.com hooks

2b7f2f62

Remove full state dict option from test_fsdp_cpu_trainable

3075d622

[pre-commit.ci] auto fixes from pre-commit.com hooks

a3e129a7

zhixiangli force pushed from 5e2f9520 to a3e129a7 24 days ago

Merge branch 'master' into feature/fsdp-cpu

72169dc4

deependujha commented on 2026-07-31

Apply suggestion from @deependujha

7bc559e9

Apply suggestion from @deependujha

0bab0a73

Apply suggestion from @deependujha

7ed3fb54

deependujha approved these changes on 2026-07-31

Merge branch 'master' into feature/fsdp-cpu

fa1458aa

Reviewers

deependujha

tchaton

justusschock

ethanwharris

Assignees

No one assigned

Labels

None yet

Milestone

No milestone