flax
Remove image from Imagenet Colab temporarily since it gave an internal problem.
#956
Merged
Merged