Simple implementation of a neural network in python from scratch using only numpy. (torch is still used to import MNIST dataset)