[build-system]
requires = ["setuptools>=68.0"]
build-backend = "setuptools.build_meta"

[project]
name = "arbitor"
version = "0.2.0"
description = "ARB (Any Relational Bit) — ternary-weighted neural network system"
requires-python = ">=3.12"
license = {text = "MIT"}
dependencies = [
    "torch>=2.5",
    "einops",
    "tqdm",
]

[project.optional-dependencies]
dev = ["pytest"]
cuda = ["torch>=2.5", "triton>=3.0"]
triton = ["triton>=3.0"]
tilelang = ["tilelang"]

[tool.setuptools.packages.find]