gpt-neoxAn implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed librariesATK6952DEF6041