torch>=2.0.0
scipy
numpy

[cuda]

[cuda:platform_system == "Linux" and platform_machine == "x86_64"]
triton<4,>=3.4

[dev]
pytest
pytest-cov
pre-commit
commitizen
torchmetrics

[test]
pytest
pytest-cov
torchmetrics
