File size: 501 Bytes
d8bc908 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | [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]
|