pip
packaging
torchvision
tqdm
torchmetrics
torch

[dev]
