flax

293b1f62 - Remove image from Imagenet Colab temporarily since it gave an internal problem.

Remove image from Imagenet Colab temporarily since it gave an internal problem. FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/google/flax/pull/907 from andsteing:imagenet 47036520906456b9b1c150854df50d10e1c675c8 PiperOrigin-RevId: 354507006