torchvision
torch
einops
omegaconf
tqdm
pytorch-lightning
torchmetrics==0.11.4