torch>=1.7.0
tqdm>=4.50.0

[dev]
pytest
flake8

[examples]
scikit-learn>=0.24.0
