torch>=1.10.0
torchvision>=0.11.1
einops
pytorch-lightning==2.0.2
torchmetrics<0.12.0,>=0.6.0
tqdm
wandb
scipy
timm
scikit-learn
hydra-core

[dali]
nvidia-dali-cuda110

[h5]
h5py

[umap]
matplotlib
seaborn
pandas
umap-learn
