Distributed TensorFlow with MPI