torch>=1.12.1
torchvision>=0.13.1
typing-extensions>=4.1.1
scikit-learn>=0.24.2

[dev]
pytest
