working on custom scheduler
mpi_dnn works in toy cases