flax

ff55c0db - Merge pull request #2983 from google:dependabot/pip/examples/imagenet/tensorflow-2.11.1