torch>=1.5.0
mpi4py
