CUDA Automatic Mixed Precision examples — PyTorch 2.3 documentation