torch>=2.0.0
numpy>=1.21.0

[all]
torch-kitsune[dev,docs,triton,viz]

[dev]
pytest>=7.0.0
pytest-benchmark>=4.0.0
black>=23.0.0
isort>=5.12.0
mypy>=1.0.0
ruff>=0.1.0
build>=1.0.0
twine>=4.0.0

[docs]
mkdocs>=1.5.0
mkdocs-material>=9.4.0
mkdocstrings[python]>=0.24.0
mkdocs-git-revision-date-localized-plugin>=1.2.0

[triton]

[triton:sys_platform == "linux"]
triton>=2.1.0

[viz]
matplotlib>=3.5.0
tensorboard>=2.10.0
