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