Mixed Precision Training
Automatic differentiation in PyTorch