Automatic Mixed Precision Pytorch Cuda