A

apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch