GitHub repository: nanz-nv/apex
Primary language: Python
Description: A PyTorch Extension:  Tools for easy mixed precision and distributed training in Pytorch