Yes, we have the HMP package for mixed precision training. Detailed support for Mixed Precision is in the PyTorch User Guide