Installation

QTen requires Python 3.11 or newer.

Install PyTorch

Install PyTorch first, choosing the CPU or CUDA build that matches your platform. Use the official PyTorch installation command for your system.

For a basic uv environment:

uv add torch

With uv:

uv add qten

With pip:

pip install qten

Install qten-plots if you want the registered obj.plot(...) helpers.

With uv:

uv add qten-plots

With pip:

pip install qten-plots

From a clone of this repository, create a CPU-only development environment with:

uv sync --extra cpu --group dev

For CUDA development, replace cpu with one of the CUDA extras configured by the repository:

uv sync --extra cu126 --group dev
uv sync --extra cu128 --group dev
uv sync --extra cu129 --group dev
uv sync --extra cu130 --group dev

To include the documentation toolchain as well:

uv sync --extra cpu --group dev --group docs