Hi jxm, I meet a very similar case with you that failing with slurm… Any solution here? What’s your github issue?