PMIx based wire-up for PyTorch