nats_bench>=1.4
torchvision
pytorch/pytorch