CNNs for MNIST

Simple CNN implementation for MNIST digit classification using PyTorch, using Apple MPS backend.