Automatic Mixed Precision examples — PyTorch 2.7 documentation
pytorch.org