MegEngine/imperative/python/megengine/amp/__init__.py

import mprop

from ..core.tensor.amp import *
from .autocast import autocast
from .convert_format import convert_module_format, convert_tensor_format
from .grad_scaler import GradScaler

mprop.init()