unsloth-zoo
Utils for Unsloth
What it is and what it does
Unsloth Zoo is a utility package that provides optimized training recipes for large language models, integrating with PyTorch, transformers, and peft to enable memory-efficient fine-tuning. It applies techniques like quantization and gradient checkpointing to reduce GPU memory requirements and accelerate training. The package supports various model architectures and works with both consumer and enterprise GPUs.
Users typically load a pretrained model, apply optimization techniques, and train on custom datasets using the trl library for instruction tuning or reasoning tasks. The package includes support for distributed training through accelerate and can export fine-tuned models to various formats. It depends on 27 runtime packages including torch, transformers, trl, peft, and accelerate.
Use it for:
- Fine-tune language models on consumer GPUs using quantization and memory-efficient kernels.
- Train reasoning models with reduced VRAM by combining gradient checkpointing and optimized operations.
- Adapt vision models for custom image-text tasks with reduced memory footprint.
- Export fine-tuned models to GGUF or Ollama format for local deployment after training.
- Implement text-to-speech model fine-tuning using memory-efficient infrastructure.
- Perform long-context training on standard GPUs by combining optimization techniques.
Worth the install?
AI-flagged interpretation of the facts on this page — verify before relying
Unsloth Zoo provides utilities for fine-tuning large language models with reduced memory usage and faster training speeds.
Yes, if you are fine-tuning large language models and want to reduce GPU memory usage and training time. The package is actively maintained with no known vulnerabilities and integrates with the standard Hugging Face ecosystem. The LGPL-3.0-or-later license is permissive for research and commercial use provided you distribute source code. Install friction is low. Verify that the specific models and hardware you plan to use are supported before committing to a large training run.
Install
unsloth-zoo on PyPI
pip
pip install unsloth-zoouv
uv add unsloth-zoopoetry
poetry add unsloth-zooInstalling unsloth-zoo
Before you install
Low install friction with a pure Python wheel. Active maintenance with recent releases and a large repository (71439 stars). Depends on 27 runtime packages including torch, transformers, and trl, which are standard in the ML ecosystem.
License in practice
Licensed under LGPL-3.0-or-later (copyleft). Derivative works and modifications must be distributed under compatible terms; commercial use is permitted if source code and license are provided.
Quickstart
pip install unsloth-zoo
from unsloth import FastLanguageModel
import torch
model, tokenizer = FastLanguageModel.from_pretrained(
model_name="unsloth/model",
load_in_4bit=True,
)
model = FastLanguageModel.get_peft_model(
model,
target_modules=["q_proj", "v_proj"],
bias="none",
use_gradient_checkpointing="unsloth",
)
Requires PyTorch and CUDA-capable GPU or compatible accelerator; on Windows, PyTorch must be pre-installed before installing unsloth-zoo.
Verify before relying
- Exact speedup and memory reduction percentages claimed in description for specific model/hardware combinations.
- Whether all listed model families (gpt-oss, Gemma 3n, Qwen3, Llama 4, Mistral) are fully supported in version 2026.8.12.
- Compatibility with non-NVIDIA GPUs beyond the Blackwell mention.
- Specific LoRA configuration parameters and default values supported by this version.
Package facts
| License | LGPL-3.0-or-later (copyleft) |
| Python support | supports the current Python release (<3.15,>=3.9) |
| Install friction | low — pure-Python wheel |
| Runtime dependencies | 27 — torch, torchao, triton, tyro, accelerate, trl, peft, cut_cross_entropy, mlx, mlx-lm, mlx-vlm, packaging, transformers, datasets, sentencepiece, tqdm, psutil, wheel, numpy, protobuf, huggingface_hub, hf_transfer, pillow, regex, msgspec, typing_extensions, filelock |
| Maintenance | actively maintained — 0 days since the last release |
| Last repo commit | |
| First released | |
| Downloads | 1,352,315/month — #4,012 on PyPI (30-day window, as of 2026-08-14) |
| Known vulnerabilities | none known (OSV.dev, checked 2026-08-14) |
Evidence: unsloth_zoo-2026.8.12-py3-none-any.whl
Keywords: ai, llm
Tags
More Artificial Intelligence packages
LiteLLM provides a unified Python interface to…
permissive · top 100 on PyPI
huggingface-hubClient library and CLI tool for downloading,…
permissive · top 100 on PyPI
langchainLangChain provides a framework for building…
permissive · top 1,000 on PyPI
hf-xethf-xet provides chunk-based deduplication and…
permissive · top 1,000 on PyPI
tokenizersTokenizers converts raw text into token…
permissive · top 1,000 on PyPI
transformersTransformers provides a unified framework for…
permissive · top 1,000 on PyPI
llamafactoryLLaMA Factory provides a unified framework for…
permissive · top 15,000 on PyPI
unslothUnsloth accelerates training and fine-tuning of…
permissive · top 5,000 on PyPI
torchaoTorchAO applies quantization and sparsity…
unclear · top 5,000 on PyPI
torchtunetorchtune is a PyTorch library for fine-tuning,…
permissive · top 15,000 on PyPI
ms-swiftms-swift is a framework for training,…
permissive · top 15,000 on PyPI
llama-modelsProvides a CLI and Python interface to…
unclear · top 15,000 on PyPI
peftPEFT implements parameter-efficient fine-tuning…
permissive · top 5,000 on PyPI
nvidia-modeloptApplies state-of-the-art model optimization…
permissive · top 15,000 on PyPI
liger-kernelLiger Kernel provides optimized Triton kernels…
permissive · top 15,000 on PyPI
transformer-engineTransformer Engine accelerates Transformer…
unclear · top 15,000 on PyPI