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