flax

Updates imagenet,mnist examples with result tables

#708

Merged