If you're fine-tuning LLMs with Axolotl, this skill gives you direct guidance on YAML configs, LoRA/QLoRA setups, and alignment techniques like DPO and ORPO. It covers practical stuff like FSDP configuration, context parallelism settings, and NCCL performance testing. The quick reference includes actual config snippets and the references folder has detailed docs on dataset formats and API usage. Honestly, Axolotl's YAML-based approach is refreshingly straightforward compared to writing training loops from scratch, and having the multimodal support patterns documented here saves you from piecing together examples from GitHub issues.

skills.sh installs
1K
GitHub stars
14.4K
last updated
Aug 21, 2026

Install

$ npx -y skills add orchestra-research/ai-research-skills --skill axolotl --agent claude-code

Installs into .claude/skills of the current project.

Files

Axolotl Skill

Comprehensive assistance with axolotl development, generated from official documentation.

When to Use This Skill

This skill should be triggered when:

  • Working with axolotl
  • Asking about axolotl features or APIs
  • Implementing axolotl solutions
  • Debugging axolotl code
  • Learning axolotl best practices

Quick Reference

Common Patterns

Pattern 1: To validate that acceptable data transfer speeds exist for your training job, running NCCL Tests can help pinpoint bottlenecks, for example:

./build/all_reduce_perf -b 8 -e 128M -f 2 -g 3

Pattern 2: Configure your model to use FSDP in the Axolotl yaml. For example:

fsdp_version: 2
fsdp_config:
  offload_params: true
  state_dict_type: FULL_STATE_DICT
  auto_wrap_policy: TRANSFORMER_BASED_WRAP
  transformer_layer_cls_to_wrap: LlamaDecoderLayer
  reshard_after_forward: true

Pattern 3: The context_parallel_size should be a divisor of the total number of GPUs. For example:

context_parallel_size

Sponsors

Details

Repository
github.com/orchestra-research/ai-research-skills
Categories
AI & Agent Building, Data Science & ML
First seen
Jun 3, 2026

More from orchestra-research/ai-research-skills

All 95
1orchestra-research avatarhqq-quantizationorchestra-research/ai-research-skillsHalf-Quadratic Quantization for LLMs without calibration data. Use when quantizing models to 4/3/2-bit precision without needing calibration datasets, for fast 1Kskills.sh installs
2orchestra-research avatarhuggingface-tokenizersorchestra-research/ai-research-skillsFast tokenizers optimized for research and production. Rust-based implementation tokenizes 1GB in <20 seconds. Supports BPE, WordPiece, and Unigram algorithms. 1Kskills.sh installs
3orchestra-research avatarlong-contextorchestra-research/ai-research-skillsExtend context windows of transformer models using RoPE, YaRN, ALiBi, and position interpolation techniques. Use when processing long documents (32k-128k+ token1Kskills.sh installs
4orchestra-research avatarnanogptorchestra-research/ai-research-skillsEducational GPT implementation in ~300 lines. Reproduces GPT-2 (124M) on OpenWebText. Clean, hackable code for learning transformers. By Andrej Karpathy. Perfec1Kskills.sh installs
5orchestra-research avatarpytorch-fsdp2orchestra-research/ai-research-skillsAdds PyTorch FSDP2 (fully_shard) to training scripts with correct init, sharding, mixed precision/offload config, and distributed checkpointing. Use when models1Kskills.sh installs
6orchestra-research avatarmodal-serverless-gpuorchestra-research/ai-research-skillsServerless GPU cloud platform for running ML workloads. Use when you need on-demand GPU access without infrastructure management, deploying ML models as APIs, o1Kskills.sh installs