flax

bd30f762 - Merge pull request #1092 from brettkoonce:imagenet-readme-tweaks