pyproject.toml

[tool.poetry]
name = "torch_semiring_einsum"
version = "1.2.0"
description = "Extensible PyTorch implementation of einsum that supports multiple semirings"
license = "MIT"
authors = ["Brian DuSell <bdusell@gmail.com>"]
readme = "README.md"
homepage = "https://bdusell.github.io/semiring-einsum/"
repository = "https://github.com/bdusell/semiring-einsum"
classifiers = [
    "Operating System :: OS Independent",
    "Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Scientific/Engineering :: Mathematics"
]

[tool.poetry.dependencies]
python = "^3.6.1"
pynvml = "^11.0.0"
torch = ">=1.1.0, <3"
typing-extensions = "^4.0.0"

[tool.poetry.dev-dependencies]
sphinx = "^5.1.1"
sphinxcontrib-bibtex = "^2.5.0"
setuptools = { version = "^70.1.1", python = ">=3.8" }
sphinx-autodoc-typehints = { version = "^1.19.2", python = ">=3.7" }
numpy = { version = "^1.23", python = ">=3.8" }
matplotlib = { version = "^3.5", python = ">=3.7" }
pynvml = { version = "^11.4.1", python = ">=3.6" }
torch = { version = "^2.3.0", python = ">=3.8" }
typing-extensions = { version = "^4.12.0", python = ">=3.8" }

[build-system]
requires = ["poetry_core>=1.0.0"]
build-backend = "poetry.core.masonry.api"