# pyrefly: ignore [deprecated] from .autocast_mode import autocast from .grad_scaler import GradScaler