Multi-node, multi-GPU training with PyTorch Lightning on SageMaker - View it on GitHub