numpy
pyyaml
torch
torchvision
matplotlib
hyperopt
