torch
torchvision

nats_bench>=1.4
nas_201_api