flax

15b6229d - Merge pull request #1113 from brettkoonce:mnist-readme-tweak