mpi4py
numpy
tensorflow
