GitHub repository: microsoft/deepspeed-gpt-neox
Description: An implementation of model parallel autoregressive transformers on GPUs, based on the DeepSpeed library.