single-GPU to multi-GPU training of PyTorch apps at NERSC - View it on GitHub