Search tools, frameworks, libraries, categories and domains
Distributed training library with ZeRO memory optimizations for training very large models.
pip install deepspeed
Package deepspeed, last 26 weeks.
deepspeed
Other tools in the same categories.
unslothai/unsloth
Fast, memory-efficient LoRA/QLoRA fine-tuning and RL for open LLMs on a single consumer GPU.
hiyouga/LlamaFactory
Unified fine-tuning toolkit with a web UI supporting 100+ LLMs and VLMs across SFT, DPO and more.
huggingface/trl
Hugging Face library for post-training LLMs with SFT, DPO, GRPO and other RL-based methods.
axolotl-ai-cloud/axolotl
Config-driven fine-tuning framework for LLMs supporting full, LoRA and QLoRA training at multi-GPU scale.
verl-project/verl
ByteDance's flexible reinforcement learning library for LLM post-training (PPO, GRPO) at scale.
NVIDIA/Megatron-LM
NVIDIA's reference for training transformer models at massive scale with tensor and pipeline parallelism.
pytorch/torchtitan
PyTorch-native platform for large-scale LLM pretraining with FSDP2, tensor and pipeline parallelism.
bitsandbytes-foundation/bitsandbytes
8-bit and 4-bit quantization kernels that make QLoRA and low-memory inference possible.
thinkingmachines.ai
Thinking Machines' managed API for fine-tuning open-weight models while you keep control of the training loop.