MNIST: Design-Your-Own tiny Pytorch Model