matplotlib
numpy
PyYAML
setuptools
tensorflow>=2.15.0

[parallel]
horovod>=0.28.1
