DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

A simple, performant and scalable Jax LLM!

A Native-PyTorch Library for LLM Fine-tuning.

veRL is a flexible and efficient RL framework for LLMs.

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.

Ongoing research training transformer models at scale.