flax

a7b0242d - Merge pull request #2980 from google:dependabot/pip/examples/sst2/tensorflow-2.11.1