* Re-align docs with code example for mnist * Also fix dist_mnist and add context highlight
Attach a file by drag & drop or click to upload
