Automatic Mixed Precision Pytorch Documentation