axolotl

library0.17.0pypypiunverified

A user-friendly LLM training framework supporting a wide range of models, including Llama, Mistral, and others. Axolotl streamlines fine-tuning with PEFT, FSDP, and other optimizations, currently at version 0.16.1. Release cadence is regular, with multiple releases per month.

llm-agents ai-ml

pip install axolotl

github ↗package ↗homepage ↗docs ↗

INSTALL

IMPORT

SIG · AXOLOTL

axolotl

llm-agentspythonv0.17.0

Install

—

Import

—

Disk

—

Pass rate

0/ 10

Env Coverage0 / 10

glibc

3.9–3.13

musl

3.9–3.13

Install & Compatibility

Where this runs

tested against v? · pip install

Install × environment matrix

Each cell = how many times install + import succeeded across repeated harness runs. Partial = flaky.
glibc = Debian/Ubuntu slim · musl = Alpine Linux

musl

py 3.10–3.915 runs

build_error

glibc

py 3.10–3.915 runs

timeout

Code

Verified usage

Verified import paths — ran on the pinned version, not inferred.

train

✓ from axolotl.utils.trainer import train

✗ from axolotl.utils.trainer import train

version

✓ from axolotl import version

✗ from axolotl import version

PackageNotFoundError

✓ from axolotl import PackageNotFoundError

✗ from axolotl import PackageNotFoundError

Basic training using a YAML config file.

from axolotl.utils.config import load_config from axolotl.utils.trainer import train config = load_config('config.yml') # config must include 'model_name_or_path' and other required fields train(cfg=config)

axolotl --version

Debug

Known issues

breakingVersion 0.4.0 changed the trainer API: train() now requires a config object, not kwargs.

fix

Use load_config to load a YAML or dict config, then call train(cfg=config).

affects: >=0.4.0

gotchaFlash Attention must be installed separately and CUDA-compatible GPU required.

fix

Install with pip install axolotl[flash-attn] and ensure GPU is available.

affects: all

gotchaYAML config keys are case-sensitive and must match exactly with axolotl's schema.

fix

Refer to the official config documentation for correct key names.

affects: all

deprecatedThe cli.py entry point is deprecated; use the Python API or `axolotl` command-line tool.

fix

Use `python -m axolotl.cli.train config.yml` or the Python API.

affects: >=0.5.0

Upgrade

Version history

0.17.0latest on PyPI · released Jun 3, 2026

Audit

Dependencies

torchrequiredCore dependency for model operations

transformersrequiredHugging Face transformers library

datasetsrequiredDataset loading and processing

acceleraterequiredTraining acceleration

peftoptionalParameter-Efficient Fine-Tuning

deepspeedoptionalDeepSpeed integration for distributed training

flash-attnoptionalFlash Attention for memory-efficient attention

Agent activity

28 hits · last 30 days

node

Amazon

OpenAI (training)

Resources

githubhttps://github.com/axolotl-ai-cloud/axolotl.git ↗

packagehttps://pypi.org/project/axolotl/ ↗

homepagehttps://axolotl.ai/ ↗

docshttps://docs.axolotl.ai/ ↗