torchtnt>=0.0.4
typing_extensions

[dev]
numpy
torchvision
pre-commit
pytest
pytest-timeout
pytest-cov
Cython>=0.28.5
scikit-learn==0.22
