Bump flax from 0.10.1 to 0.10.2 (#111)

Bumps [flax](https://github.com/google/flax) from 0.10.1 to 0.10.2.
- [Release notes](https://github.com/google/flax/releases)
- [Changelog](https://github.com/google/flax/blob/main/CHANGELOG.md)
- [Commits](google/flax@v0.10.1...v0.10.2)

---
updated-dependencies:
- dependency-name: flax
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

pyproject.toml

@@ -16,7 +16,7 @@ keywords = []
 # pip dependencies of the project
 dependencies = [
     "jax==0.4.35",
-    "flax==0.10.1",
+    "flax==0.10.2",
     "ml_dtypes==0.4.0",
     "optax==0.2.4",
     "orbax-checkpoint==0.9.1",