LLM fine tuning made easy
Axolotl streamlines your environments, architecture, datasets, configurations, GPU access and much more!
Supported LLMs
- Mistral AI
- llama
- Eleuther AI
- Falcon
- MPT
- HuggingFace
- Cerebras
- XGen
- Qwen
- RWKV
- Gemma
- MS Phi
Deploy optimized LLMs in a few clicks
Train
Train various Huggingface models such as llama, pythia, falcon, mpt.
Full Finetune
Supports full finetune, lora, qlora, relora, and gptq.
Configuration
Customize configurations & hyperparameters using a simple yaml file or CLI overwrite.
Datasets
Load different dataset formats, use custom formats, or bring your own tokenized datasets.
Integrations
Integrated with xformer, flash attention, rope scaling, and multipacking.
Multi GPU
Works with single GPU or multiple GPUs via FSDP or Deepspeed.
Built with Axolotl
Community showcase
Cloud & GPU Partners
- Deepspeed
- Vllm
- Skypilot
- Runpod
- Wandb.ai
- Jarvislabs.ai
- LambdaLabs
- Latitude.sh