I am trying to train my model in a distributed fashion with multiple nodes using a SLURM cluster in an HPC environment. Therefore, I would like to use the MultiWorkerMirroredStr
