A Native-PyTorch Library for LLM Fine-tuning.

An implementation of model parallel autoregressive transformers on GPUs, based on the DeepSpeed library.

Mesh TensorFlow: Model Parallelism Made Easier.

DeepSpeed version of NVIDIA's Megatron-LM that adds additional support for several features such as MoE model training, Curriculum Learning, 3D Parallelism, and others.

Efficient Training for Big Models.

Ongoing research training transformer models at scale.