skillfed

unsloth-zoo

Utils for Unsloth

unsloth-zoo v2026.8.12 1.4M downloads/30d#4,012 on PyPI71,439
Copyleft license LGPL-3.0-or-later Active released

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-zoo

uv

uv add unsloth-zoo

poetry

poetry add unsloth-zoo

Installing 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

Programming Language :: Python

Tags

llm fine-tuning optimizationreduce vram gpu traininglanguage model memory efficientfaster model training utilsquantization fine-tunemodel optimizationgpu memory reduction training
model-optimizationgpu-memory-efficientfine-tuning

More Artificial Intelligence packages

Further reading