https://pytorch.org/docs/stable/distributed.html#torch.distributed.init_process_group
torch.distributed.init_process_group(backend=None, init_method=None, timeout=datetime.timedelta(seconds=1800), world_size=- 1, rank=- 1, store=None, group_name='', pg_options=None)