Lightning-AI/litgpt
A high-performance, no-abstraction toolkit providing recipes for pretraining, finetuning, and deploying over 20 large language models at scale.
Core Features
Quick Start
pip install 'litgpt[extra]'Detailed Introduction
LitGPT is an open-source toolkit from Lightning AI designed to empower developers with high-performance, enterprise-ready solutions for large language models. By providing over 20 LLM implementations built from scratch with minimal abstractions, it offers unparalleled control and ease of debugging. The project includes highly optimized recipes for pretraining, finetuning (supporting techniques like LoRA and QLoRA), and deploying LLMs at scale, leveraging advanced features like Flash Attention and FSDP for superior performance and memory efficiency across diverse hardware setups, from single GPUs to large clusters.