mnist.py

https://github.com/cvxgrp/pymde/blob/main/examples/mnist.py