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