Unsloth Stubs

Unsloth Stubs is a collection of PEP 484 type stub files (.pyi) for the unsloth library. These stubs provide accurate type hints for improved autocompletion, static analysis, and developer experience in Python IDEs and type checkers.

Stub files are generated using stubgen and manually refined for correctness. Some tests and configs are contributed to Typeshed; see their contributors and license.

Features

• Accurate type hints for Unsloth modules
• Enhanced autocompletion in editors
• Compatibility with Mypy and other type checkers
• Easy installation via PyPI or Conda

Installation

Install from PyPI:

pip install unsloth-stubs

Or from conda-forge:

conda install -c conda-forge unsloth-stubs

Usage

Once installed, type checkers (e.g., Mypy, Pytype¹) and autocompletion tools (e.g., Jedi) will automatically use the stubs if they are on your PYTHONPATH. If you need to manually add stubs, copy the .pyi files next to their .py counterparts in your Unsloth installation directory, or add the stubs directory to your PYTHONPATH.

Refer to PEP 561 for details on distributing and packaging type information.

Compatibility

Unsloth Stubs versions follow Unsloth releases. For example, unsloth-stubs==2025.9.5 is compatible with unsloth>=2025.9.5. Maintenance releases (e.g., post1, post2, ...) are reserved for annotation updates only.

API Coverage

The stubs cover the most commonly used classes and methods in Unsloth. Full coverage is not guaranteed, but contributions are welcome!

Project Structure

unsloth-stubs/
├── unsloth/
│   ├── __init__.pyi
│   ├── models/
│   ├── kernels/
│   ├── dataprep/
│   ├── registry/
│   └── utils/
└── ...

Contributing

Contributions, bug reports, and suggestions are welcome! Please open an issue or pull request on GitHub.

License & Disclaimer

Unsloth and the Unsloth logo are trademarks of their respective owners. This project is not affiliated with or endorsed by any other organization.