flax

752a2c07 - Merge pull request #1362 from ilopezgp:spelling_mnist