Fix #2826: implement gradient checkpoint callbacks#2860

Open

githubnemo wants to merge 7 commits intohuggingface:mainfrom

githubnemo:issue/fix-gradient-checkpointing

Commits on Oct 22, 2025

Remove gradient requirement in CPT's embedding
nemo
committed
Improve gradient checkpointing tests
nemo
committed
Fix #2826: implement gradient checkpoint callbacks
nemo
committed
Make style
nemo
committed

Commits on Oct 23, 2025

Address test flakyness
nemo
committed

Commits on Oct 29, 2025

Add support for DPO training
nemo
committed
Address review comments
nemo
committed