build_dependencies:
  - pip
dependencies:
  - torch
  - torchvision
  - mlflow
  - tensorboardX