update the MNIST ipynb to use the new runner

[skip ci]