flax

2248f44b - Copybara import of the project:

Copybara import of the project: -- 5b4f79feb0bc7d4d17eff96a1dfe53d000c7d1f6 by dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>: Bump tensorflow from 2.9.3 to 2.11.1 in /examples/wmt Bumps [tensorflow](https://github.com/tensorflow/tensorflow) from 2.9.3 to 2.11.1. - [Release notes](https://github.com/tensorflow/tensorflow/releases) - [Changelog](https://github.com/tensorflow/tensorflow/blob/master/RELEASE.md) - [Commits](https://github.com/tensorflow/tensorflow/compare/v2.9.3...v2.11.1) --- updated-dependencies: - dependency-name: tensorflow dependency-type: direct:production ... Signed-off-by: dependabot[bot] <support@github.com> COPYBARA_INTEGRATE_REVIEW=https://github.com/google/flax/pull/2981 from google:dependabot/pip/examples/wmt/tensorflow-2.11.1 5b4f79feb0bc7d4d17eff96a1dfe53d000c7d1f6 PiperOrigin-RevId: 519691378