PR #48444 Add offload to gradient checkpointing

Add offload to gradient checkpointing #48444

ArthurZucker merged 2 commits into main from activation-checkpointing-offload

Add offload to gradient checkpointing

7676358a

Follow the repo conventions for the new option

d9325472

ArthurZucker requested a review from

3outeille 34 days ago

ArthurZucker approved these changes on 2026-09-01

ArthurZucker merged 69a7fb1a into main 34 days ago

ArthurZucker deleted the activation-checkpointing-offload branch 34 days ago

Reviewers

ArthurZucker

3outeille

Assignees

No one assigned

Labels

None yet

Milestone

No milestone