Installation

You need to install triton and torch on your system or create a venv with:

# With uv 
uv sync

# Or manually
python -m venv .venv
source .venv/bin/activate
pip install build
pip install -e .

Then, set the Rust nightly toolchain and build:

rustup toolchain install nightly
cargo run

The cargo build hooks will take care of compiling the PTX files with Python.