Simple PyTorch Distributed Training Example - View it on GitHub