phivenv/Lib/site-packages/torch/cpu/amp/__init__.py

from .autocast_mode import autocast
from .grad_scaler import GradScaler