Add benchmark script for MNIST.