numpy
opt-einsum
torch<3,>=2.4
typing-extensions

[dev]
black
covdefaults
mypy
pre-commit
pylint
pytest
pytest-cov
pytest-random-order
pytest-xdist
scipy
tox

[tox]
covdefaults
pytest
pytest-cov
pytest-random-order
pytest-xdist
scipy

[triton]
triton>=2.0
