Pillow
numpy
optuna
scikit-learn
scikit-image
timm==0.8.19.dev0
torch>=1.13.1
torchvision
tqdm
scipy
accelerate
psutil
pandas
wilds
faiss-cpu
matplotlib

[dev]
black
isort
flake8
pytest
pytest-cov
pytest-timeout
pytest-xdist
twine
sphinx
sphinx_rtd_theme
myst-parser
six
