Stars
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch
Code for "Heterogeneous Graph Transformer" (WWW'20), which is based on pytorch_geometric