torch
torchvision
ray
torchinfo
torchmetrics
tqdm
numpy
