NVIDIA / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
8,392Updated last week

Related projects

Alternatives and complementary repositories for apex