torch>=2.2.0
numpy>=1.22.0
tqdm
opencv-python
Pillow
scipy
geomloss
joblib
sympy
scikit-learn
seaborn
einops>=0.8.0

[benchmark]
torchvision
typer
pynndescent>=0.5.13
pytorch-lightning
PyYAML
matplotlib
pandas
timm>=1.0.14
lightly>=1.5.18
