numpy>=1.18
scikit-learn>=1.0

[all]
torch>=1.3.1
torchmetrics>=0.2
tensorflow>=2.0

[keras]
tensorflow>=2.0

[pytorch]
torch>=1.3.1
torchmetrics>=0.2

[tests]
pytest
pytest-cov
pytest-xdist
