matplotlib
numpy
PyYAML
setuptools
tensorflow

[parallel]
horovod>=0.28.1
