pyproject.toml

[build-system]
requires = ["setuptools", "Cython"]
build-backend = "setuptools.build_meta"

[project]
name = "tvm"
version = "0.16.0"
requires-python = ">=3.10"
description = "TVM: An End to End Tensor IR/DSL Stack for Deep Learning Systems"
authors = [{ name = "Apache TVM" }]
license = { file = "LICENSE" }
# See https://pypi.org/classifiers/
classifiers = [
  "License :: OSI Approved :: Apache Software License",
  "Development Status :: 4 - Beta",
  "Intended Audience :: Developers",
  "Intended Audience :: Education",
  "Intended Audience :: Science/Research",
]
keywords = ["machine learning"]

dependencies = [
  "attrs ~= 23.2",
  "cloudpickle ~= 3.0",
  "decorator ~= 5.1",
  "ml-dtypes ~= 0.3",
  "numpy ~= 1.26",
  "psutil ~= 5.9",
  "scipy ~= 1.12",
  "tornado ~= 6.4",
  "typing-extensions ~= 4.10",
  "beartype ~= 0.17",
]

[project.optional-dependencies]
pytorch = ["torch ~= 2.2", "torchvision ~= 0.17"]
test = ["pytest ~= 8.1"]
highlight = ["pygments ~= 2.17", "packaging ~= 24.0"]

[tool.isort]
profile = "black"
src_paths = ["python", "tests/python"]

[tool.black]
line-length = 100
target-version = ['py310']
include = '(\.pyi?$)'