megengine.amp¶

`GradScaler`	A helper class that performs grad scaling to prevent from data overflow in `autocast` mode.
`autocast`	A class to control autocast mode for amp as a context manager or a decorator.
`convert_module_format`	Convert NCHW Module to NHWC Module.
`convert_tensor_format`	Convert NCHW Tensor to NHWC Tensor.

`enabled`	获取或设置是否启用 amp autocast.
`high_prec_dtype`	Get or set amp autocast mode's higher precision dtype.
`low_prec_dtype`	Get or set amp autocast mode's lower precision dtype.

GraphOptimizationConfig

GradScaler