torch>=2.0.0
numpy>=1.21.0
packaging>=21.0

[all]
torch-kitsune[dev,docs,hf,memory,predict,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

[hf]
transformers>=4.30.0

[memory]
supermemory>=3.0.0

[predict]
click>=8.0
rich>=13.0

[tensorrt]
onnx>=1.14.0
onnxsim>=0.4.33

[tensorrt:sys_platform == "linux"]
tensorrt>=8.6.0
pycuda>=2022.1

[triton]

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

[viz]
matplotlib>=3.5.0
tensorboard>=2.10.0
