import mprop from ..core.tensor.amp import * from .autocast import autocast from .grad_scaler import GradScaler mprop.init()