numpy~=1.20.2
scipy~=1.6.2
scikit-learn~=0.24.1
setuptools~=54.2.0
pandas~=1.2.3
matplotlib~=3.4.1
torchmetrics~=0.7.2

[doc]
sphinx

[test]
pytest>5.0.0
