refactor(amp): migrate from torch.cuda.amp to torch.amp#3436

Closed

keelobytes wants to merge 2 commits intopytorch:masterfrom

keelobytes:fix-future-warnings

Commits on Jul 26, 2025

refactor(amp): migrate from torch.cuda.amp.GradScaler to torch.amp.GradScaler

Keerthana Srinivasa Naidu
authored and
Keerthana Srinivasa Naidu
committed
fix(docs): suppress unresolved reference to torch.amp.grad_scaler.GradScaler
keelobytes
committed