pytorch-ignite
ray[tune]
torch
torchvision
