Add --mpi=pmix to COMMAND srun in Ray Slurm template#439

athitten wants to merge 2 commits intomainNVIDIA-NeMo/Run:mainfrom

athitten/raytemplate_mpiNVIDIA-NeMo/Run:athitten/raytemplate_mpiCopy head branch name to clipboard