The PyTorch MNIST example as implemented in the [PyTorch
repository](https://github.com/pytorch/examples/blob/main/mnist/main.py).