PyTorch examples for NERSC systems - View it on GitHub