Automatic Mixed Precision examples — PyTorch 2.11 documentation
pytorch.org