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