torch
torchvision
numpy
ax-platform
mlflow
haikunator
