Repository navigation

[PyTorch] Preserve CUDA RNG state during checkpoint recompute - #3835

Open

losewayy wants to merge 2 commits into

NVIDIA:mainfrom

losewayy:fix/cudarngstates-tracker-checkpoint-dropout

Open

[PyTorch] Preserve CUDA RNG state during checkpoint recompute#3835
losewayy wants to merge 2 commits into
NVIDIA:mainfrom
losewayy:fix/cudarngstates-tracker-checkpoint-dropout

Commits on Nov 8, 2026
- [PyTorch] Preserve CUDA RNG state during checkpoint recompute
  losewayy
  committed
  3816a26
- [PyTorch] Test CUDA RNG restoration after checkpoint backward
  losewayy
  committed
  613a449