Installation

Install from PyPI using uv (recommended):

uv add torchscalers

Or with pip:

pip install torchscalers

PyTorch (torch>=2.0) is the only runtime dependency and is installed automatically.

Example scripts

The example scripts require Lightning. Install the optional examples group to pull it in:

uv sync --extra examples