An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries - View it on GitHub