Copybara import of the project:
Copybara import of the project:

--
20503e8414ff2028ca04da410a117cac5f51dc3e by dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>:

Bump tensorflow from 2.8.0 to 2.8.1 in /examples/sst2

Bumps [tensorflow](https://github.com/tensorflow/tensorflow) from 2.8.0 to 2.8.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.8.0...v2.8.1)

---
updated-dependencies:
- dependency-name: tensorflow
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <5bdcd3c0d4d24ae3e71b3b452a024c6324c7e4bb@github.com>
COPYBARA_INTEGRATE_REVIEW=https://github.com/google/flax/pull/2147 from google:dependabot/pip/examples/sst2/tensorflow-2.8.1 20503e8414ff2028ca04da410a117cac5f51dc3e
PiperOrigin-RevId: 450854645

google/flax,google/flax