Computational framework to instantiate multiple groups of processes with Torch.Distributed - View it on GitHub