only use 2 MPI processes when tested, since GROMACS required one GPU per MPI process