horovod==0.28.1
matplotlib
numpy
PyYAML
setuptools
tensorflow
