mnist_from_scratch.py