torch
tqdm
ray
numpy
torchvision