Axolotl
YAML-configurable fine-tuning for LoRA, QLoRA, and full fine-tuning
๐ Overview
Axolotl is an open-source fine-tuning framework that wraps multiple approaches (LoRA, QLoRA, full fine-tuning, DPO, RLHF) in a simple YAML-configurable interface. Unlike Unsloth (which focuses on speed), Axolotl prioritizes flexibility โ supporting 50+ model architectures (Llama, Mistral, Gemma, Qwen, Phi, Falcon, MPT), multiple datasets formats, and advanced techniques like flash attention, gradient checkpointing, and distributed training. The YAML config makes experiments reproducible and shareable, and the active community maintains templates for most popular models.
โจ Key Features
- โข12K+ GitHub stars โ most flexible fine-tuning framework
- โขYAML-configurable: model, dataset, method in one file
- โข50+ model architectures supported
- โขLoRA, QLoRA, full fine-tuning, DPO, RLHF
- โขFlash attention 2, gradient checkpointing, FSDP
- โขBuilt-in GGUF export for local inference
- โขActive community with model-specific templates
๐ฏ The Problem It Solves
Fine-tuning LLMs requires stitching together multiple libraries (PEFT, bitsandbytes, transformers, datasets) with careful memory management. Axolotl collapses this complexity into a single YAML config โ define your model, dataset, and method, then run.
๐ง How It Works
You write a YAML config specifying base model, dataset, fine-tuning method (LoRA/QLoRA/full), and training parameters. Axolotl handles the rest: model loading with quantization, dataset preprocessing, training with HuggingFace Trainer, and adapter merging. It supports flash attention 2, gradient checkpointing, and distributed training via FSDP. The framework auto-detects your hardware and adjusts settings. Export to GGUF for local inference is built-in.
๐ Installation & Quick Start
Installation
See websiteQuick Start
- See documentation
โ Pros
- โขMost flexible fine-tuning framework
- โขYAML config makes experiments reproducible
- โขBroadest model support (50+ architectures)
- โขMultiple methods: LoRA, QLoRA, DPO, RLHF
- โขBuilt-in GGUF export
- โขApache 2.0 licensed โ completely free
- โขActive community with templates
โ Cons
- โขNot as fast or memory-efficient as Unsloth
- โขRequires more GPU RAM for equivalent models
- โขYAML config has a learning curve
- โขDocumentation assumes ML expertise
- โขBreaking changes between versions
๐ฌ Practitioner Verdict
โAxolotl is the most flexible fine-tuning framework โ the YAML config makes experiments reproducible, and the model support is unmatched. The trade-off: not as fast or memory-efficient as Unsloth, requires more GPU RAM for equivalent models, and the YAML config has a learning curve. For researchers and teams who need maximum flexibility and model choice, Axolotl is the default.โ
Self-Hosted (Free)
Open source, MIT/Apache licensed. Run it yourself.
โญ Star & Clone on GitHubFree forever. Your infrastructure, your data.
Deployment Options
Ways to run Axolotl in production
๐ Specifications
- Language
- Python
- License
- Apache-2.0
- Platform
- Linux, macOS, Windows
- Supported Models
- REST API, CLI
๐ฐ Pricing Reality
100% free Apache 2.0. No paid tier. You supply your own GPU hardware or cloud instance.