Pytorch includes a simple neural network example for the MNIST data: https://github.com/pytorch/examples/blob/main/mnist/main.py

It only takes a few minutes to train with default parameters and will have >99% accuracy on the MNIST test set.