torch
kornia
omegaconf
pytorch-lightning
torchvision
tqdm
torchmetrics
scikit-learn
numpy
matplotlib
timm==0.4.12

ninja