nats_bench>=1.4
torchvision
qlib  # For some modules