torch==1.9.1
torchvision==0.10.1
einops
pytorch-lightning==1.4.7
torchmetrics>=0.5.1
lightning-bolts>=0.4.0
tqdm
wandb
scipy
timm

[dali]
nvidia-dali-cuda110

[umap]
matplotlib
seaborn
pandas
umap-learn
