NVFP4 (4-bit) MoE LoRA training is now supported via ScatterMoE (W4A16) and SonicMoE (W4A4), including adapter merge back into a plain NVFP4 checkpoint.
MoE expert quantization support (via quantize_moe_experts: true) greatly reduces VRAM when training MoE models (FSDP2 compat).
Expand older updates
2026/02:
ScatterMoE LoRA support. LoRA fine-tuning directly on MoE expert weights using custom Triton kernels.
Axolotl now has support for SageAttention and GDPO (Generalized DPO).
2026/01:
New integration for EAFT (Entropy-Aware Focal Training), weights loss by entropy of the top-k logit distribution, and Scalable Softmax, improves long context in attention.
2025/09: Axolotl now has text diffusion training. Read more here.
2025/08: QAT has been updated to include NVFP4 support. See PR.
2025/07:
ND Parallelism support has been added into Axolotl. Compose Context Parallelism (CP), Tensor Parallelism (TP), and Fully Sharded Data Parallelism (FSDP) within a single node and across multiple nodes. Check out the blog post for more info.
TiledMLP support for single-GPU to multi-GPU training with DDP, DeepSpeed and FSDP support has been added to support Arctic Long Sequence Training. (ALST). See examples for using ALST with Axolotl!
2025/06: Magistral with mistral-common tokenizer support has been added to Axolotl. See docs to start training your own Magistral models with Axolotl!
2025/05: Quantization Aware Training (QAT) support has been added to Axolotl. Explore the docs to learn more!
2025/04: Llama 4 support has been added in Axolotl. See docs to start training your own Llama 4 models with Axolotl's linearized version!
2025/03: Axolotl has implemented Sequence Parallelism (SP) support. Read the blog and docs to learn how to scale your context length when fine-tuning.
2025/03: (Beta) Fine-tuning Multimodal models is now supported in Axolotl. Check out the docs to fine-tune your own!
2025/02: Axolotl has added LoRA optimizations to reduce memory usage and improve training speed for LoRA and QLoRA in single GPU and multi-GPU training (DDP and DeepSpeed). Jump into the docs to give it a try.
2025/02: Axolotl has added GRPO support. Dive into our blog and GRPO example and have some fun!
2025/01: Axolotl has added Reward Modelling / Process Reward Modelling fine-tuning support. See docs.
✨ Overview
Axolotl is a free and open-source tool designed to streamline post-training and fine-tuning for the latest large language models (LLMs).
Features:
Multiple Model Support: Train various models like GPT-OSS, LLaMA, Mistral, Mixtral, Pythia, and many more models available on the Hugging Face Hub.
Multimodal Training: Fine-tune vision-language models (VLMs) including LLaMA-Vision, Qwen2-VL, Pixtral, LLaVA, SmolVLM2, GLM-4.6V, InternVL 3.5, Gemma 3n, PaddleOCR-VL, Muse Glimmer, and audio models like Voxtral with image, video, and audio support.
Training Methods: Full fine-tuning, LoRA, QLoRA, GPTQ, QAT (int8/int4/FP8/NVFP4/MXFP4), FP8 mixed-precision training, NVFP4/MXFP4 MoE LoRA, Preference Tuning (DPO, IPO, KTO, ORPO), RL (GRPO, GDPO), and Reward Modelling (RM) / Process Reward Modelling (PRM).
Easy Configuration: Re-use a single YAML configuration file across the full fine-tuning pipeline: dataset preprocessing, training, evaluation, quantization, and inference.
Axolotl ships with built-in documentation optimized for AI coding agents (Claude Code, Cursor, Copilot, etc.). These docs are bundled with the pip package, no repo clone needed.
# Show overview and available training methods
axolotl agent-docs
# Topic-specific references
axolotl agent-docs sft # supervised fine-tuning
axolotl agent-docs grpo # GRPO online RL
axolotl agent-docs preference_tuning # DPO, KTO, ORPO, SimPO
axolotl agent-docs reward_modelling # outcome and process reward models
axolotl agent-docs pretraining # continual pretraining
axolotl agent-docs --list # list all topics
# Dump config schema for programmatic use
axolotl config-schema
axolotl config-schema --field adapter
If you're working with the source repo, agent docs are also available at docs/agents/ and the project overview is in AGENTS.md.
Contributions are welcome! Please see our Contributing Guide for details.
📈 Telemetry
Axolotl has opt-out telemetry that helps us understand how the project is being used
and prioritize improvements. We collect basic system information, model types, and
error rates, never personal data or file paths. Telemetry is enabled by default. To
disable it, set AXOLOTL_DO_NOT_TRACK=1. For more details, see our telemetry documentation.
What is this?Axolotl is a free, open-source framework that streamlines post-training and fine-tuning of large language models — you describe your run in one YAML config and it drives the whole pipeline.
Is it a Python library I import, or a tool I run on its own?It's a standalone training tool you run on GPU hardware rather than a library you code against: install it from PyPI or Docker, write a YAML config, and launch training from the command line.
How do I get my first fine-tune running?Install via Docker (the README calls it the least error-prone route) or pip, start from an example config in the Examples directory, point it at your model and dataset, and follow the Quick Start guide.
What kind of machine do I need to run this?Python 3.11 or newer (3.12 recommended), PyTorch 2.11+, and an NVIDIA GPU (Ampere or newer for bf16 and Flash Attention) or an AMD GPU.
What's the catch — is any of it hosted for me?No, it's entirely do-it-yourself: there's no hosted service, so you train on your own GPU box or rent one from supported cloud providers like RunPod, Vast.ai or Modal.
Does it phone home while I train?Yes, opt-out telemetry is on by default and reports basic system info, model types and error rates — never personal data or file paths. Set AXOLOTLDONOTTRACK=1 to turn it off.
What models and training methods does it support?Models from the Hugging Face Hub — the README names GPT-OSS, LLaMA, Mistral, Mixtral and Pythia, with 2026 additions like Qwen3.5, Gemma 4 and GLM-4.7-Flash. Methods span full fine-tuning, LoRA/QLoRA, QAT (int8/int4/FP8/NVFP4), preference tuning (DPO, IPO, KTO, ORPO), RL (GRPO) and reward modelling, plus vision-language and audio models like Qwen2-VL and Voxtral.
Do I have to write much code to configure a run?No — one YAML configuration file drives the entire pipeline: dataset preprocessing, training, evaluation, quantization and inference. You mostly pick an example config and edit settings rather than write training code.
Where can my training datasets live?Almost anywhere: Axolotl loads datasets from your local disk, the Hugging Face Hub, or cloud storage on S3, Azure, GCP and OCI.
What if a single GPU isn't enough for my model?It scales across hardware: multi-GPU training via FSDP1/FSDP2 or DeepSpeed, multi-node runs with Torchrun or Ray, plus parallelism tricks like sequence and context parallelism for long-context training.