flax

02e2ad13 - Copybara import of the project:

Copybara import of the project: -- 82273b0b7cf242757f5dc7bd25c286500b323dcc by dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>: Bump tensorflow from 2.9.3 to 2.11.1 in /examples/mnist 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/2978 from google:dependabot/pip/examples/mnist/tensorflow-2.11.1 82273b0b7cf242757f5dc7bd25c286500b323dcc PiperOrigin-RevId: 519691373