torch/amp/__init__.py

from .autocast_mode import (
    _enter_autocast,
    _exit_autocast,
    autocast,
    is_autocast_available,
)
from .grad_scaler import GradScaler