xla

Refactor argparser to be consistent with torch.distributed

#4149

Merged