flax

f037751c - Merge pull request #2977 from google:dependabot/pip/examples/lm1b/tensorflow-2.11.1