MNIST: Design Your Own Pytorch Optimizer