A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch - View it on GitHub