nvidia/apex: A PyTorch Extension:  Tools for easy mixed precision and distributed training in Pytorch
Category: ml-dl
Language: Python