Installation¶

Tinax supports Python 3.12, 3.13, and 3.14.

Supported platforms are glibc Linux x86-64/AArch64, Windows x86-64, and Apple Silicon macOS. Tinax pins the JAX ecosystem versions it tests.

CPU¶

pip install tinax

Install the appropriate JAX distribution explicitly:

pip install "tinax[gpu]"
pip install "tinax[tpu]"

Confirm that your selected JAX extra supports your platform and accelerator runtime before installation.

Use the locked development environment from a source checkout:

uv sync --locked
uv run pytest