Updates the requirements on [flax](https://github.com/google/flax) to
permit the latest version.
<details>
<summary>Release notes</summary>
<p><em>Sourced from <a
href="https://github.com/google/flax/releases">flax's
releases</a>.</em></p>
<blockquote>
<h2>Version 0.7.0</h2>
<h2>What's Changed</h2>
<ul>
<li>delete long-unused dotgetter utility by <a
href="https://github.com/copybara-service"><code>@​copybara-service</code></a>
in <a
href="https://redirect.github.com/google/flax/pull/3156">google/flax#3156</a></li>
<li>Add sections to flax.linen toctree by <a
href="https://github.com/cgarciae"><code>@​cgarciae</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3073">google/flax#3073</a></li>
<li>update doctest requirements to use recentish jax minver by <a
href="https://github.com/levskaya"><code>@​levskaya</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3160">google/flax#3160</a></li>
<li>Update gym to gymnasium by <a
href="https://github.com/cgarciae"><code>@​cgarciae</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3133">google/flax#3133</a></li>
<li>Add lifted transforms section to Haiku migration guide by <a
href="https://github.com/cgarciae"><code>@​cgarciae</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3158">google/flax#3158</a></li>
<li>relaxed tolerance by <a
href="https://github.com/chiamp"><code>@​chiamp</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3162">google/flax#3162</a></li>
<li>fixed broken links by <a
href="https://github.com/chiamp"><code>@​chiamp</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3161">google/flax#3161</a></li>
<li>added dict migration guide by <a
href="https://github.com/chiamp"><code>@​chiamp</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3109">google/flax#3109</a></li>
<li>Update requirements, restructure files and fix formatting for VAE
example by <a
href="https://github.com/canyon289"><code>@​canyon289</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3046">google/flax#3046</a></li>
<li>Update python version support by <a
href="https://github.com/cgarciae"><code>@​cgarciae</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3168">google/flax#3168</a></li>
<li>Improve Haiku migration guide by <a
href="https://github.com/cgarciae"><code>@​cgarciae</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3169">google/flax#3169</a></li>
<li>Set default types in Flax for Orbax restoration and add
<code>restore_with_serialized_types</code> in preparation for an
upcoming change. by <a
href="https://github.com/copybara-service"><code>@​copybara-service</code></a>
in <a
href="https://redirect.github.com/google/flax/pull/3165">google/flax#3165</a></li>
<li>fix precommit issues by <a
href="https://github.com/chiamp"><code>@​chiamp</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3170">google/flax#3170</a></li>
<li>fixed incorrect reference link by <a
href="https://github.com/chiamp"><code>@​chiamp</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3167">google/flax#3167</a></li>
<li>added absltest to linen_recurrent_test by <a
href="https://github.com/chiamp"><code>@​chiamp</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3172">google/flax#3172</a></li>
<li>remove cell_size docs from RNN by <a
href="https://github.com/cgarciae"><code>@​cgarciae</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3186">google/flax#3186</a></li>
<li>0.7.0 by <a
href="https://github.com/cgarciae"><code>@​cgarciae</code></a> in <a
href="https://redirect.github.com/google/flax/pull/3187">google/flax#3187</a></li>
</ul>
<p><strong>Full Changelog</strong>: <a
href="https://github.com/google/flax/compare/v0.6.11...v0.7.0">https://github.com/google/flax/compare/v0.6.11...v0.7.0</a></p>
</blockquote>
</details>
<details>
<summary>Changelog</summary>
<p><em>Sourced from <a
href="https://github.com/google/flax/blob/main/CHANGELOG.md">flax's
changelog</a>.</em></p>
<blockquote>
<h2>0.7.0</h2>
<ul>
<li>RNNCellBase refactor.</li>
</ul>
<h2>0.6.11</h2>
<ul>
<li>Set Orbax-as-backend to be the default checkpointing method.</li>
<li>Fix setup trigger issue under sharing and transforms.</li>
<li>Add collection to self.scope.reserve(name, col) so that sow works
with the same name in different collections.</li>
<li>Minor improvements for Sequential.</li>
<li>Improve the error message in MultiHeadDotProductAttention.</li>
<li>Allow manually specifying the rng key for Dropout.</li>
<li>RNN refactor.</li>
<li>fixed missing separator for rng fold in.</li>
</ul>
<h2>0.6.10</h2>
<ul>
<li>Rudimentary quantization support: some layers can be parametrized
with custom dot_general and conv_general_dilated.</li>
</ul>
<h2>0.6.9</h2>
<ul>
<li>Depend on <code>orbax-checkpoint</code> package instead of
<code>orbax</code>.</li>
<li>Refactored setup scripts to <code>project.toml</code>.</li>
<li>Added pretty_repr utility fn.</li>
<li>Fix get_partition_spec on replicated array.</li>
<li>Updates imagenet.ipynb to use GPU Colab runtime, and fixed
config.</li>
<li>Upgrade checkpointing code to <code>jax.sharding</code>, and with
more warnings.</li>
</ul>
<h2>0.6.8</h2>
<ul>
<li>The automatic checkpoint migration was temporarily rolled back due
to legacy compatibility issues.
<ul>
<li>We still recommend you to use the <a
href="https://flax.readthedocs.io/en/latest/guides/orbax_upgrade_guide.html">upgrade
guide</a> and migrate completely to the Orbax API to ensure
stability.</li>
<li>Or alternatively, add
<code>flax.config.update('flax_use_orbax_checkpointing', True)</code> to
your project to avoid being impacted by the automatic migration
process.</li>
</ul>
</li>
<li>Added utility functions to frozen_dict api.</li>
<li>Migrated Flax away from <code>register_keypaths</code>.</li>
<li>Fixes kwargs in convert_to_graphs_tuple_fn.</li>
<li>Fixed examples in a few ways:
<ul>
<li>Bumped the TF version</li>
<li>Used latest checkpoint formats</li>
<li>Other misc fixes.</li>
</ul>
</li>
</ul>
<h2>0.6.7</h2>
<ul>
<li>New checkpoints will be saved using Orbax! Please check out <a
href="https://flax.readthedocs.io/en/latest/guides/orbax_upgrade_guide.html">upgrade
guide</a> and consider migrating completely to the Orbax API.
<ul>
<li>You could <code>flax.config.update('flax_use_orbax_checkpointing',
False)</code> to temporarily disable this migration, but note that Flax
legacy checkpointing will be removed 3 months from Mar 10, 2023.</li>
</ul>
</li>
<li>Migrating <code>FrozenDict</code> to regular dict: utility functions
now work on both.</li>
<li>Migrated Flax dataclass and <code>FrozenDict</code> to JAX pytree
keypath API.</li>
<li>Fixed pytype and improved typing for <code>Module</code></li>
<li>Fixed up uses of PyTree and PyTreeDef types.</li>
</ul>
<!-- raw HTML omitted -->
</blockquote>
<p>... (truncated)</p>
</details>
<details>
<summary>Commits</summary>
<ul>
<li><a
href="https://github.com/google/flax/commit/7e9d7087dd4684efcbff723d8e50b667b3f557bf"><code>7e9d708</code></a>
Merge pull request <a
href="https://redirect.github.com/google/flax/issues/3187">#3187</a>
from google:0.7.0</li>
<li><a
href="https://github.com/google/flax/commit/eb1c5ee2112d174a796bba27ab17088796312e70"><code>eb1c5ee</code></a>
Merge pull request <a
href="https://redirect.github.com/google/flax/issues/3186">#3186</a>
from google:fix-rnn-docs</li>
<li><a
href="https://github.com/google/flax/commit/d405953df4fb46deea0b8bb72285243195a1764b"><code>d405953</code></a>
fix changelog</li>
<li><a
href="https://github.com/google/flax/commit/047e4e82de46f08baa14894be69b61280ee1a3ac"><code>047e4e8</code></a>
0.7.0</li>
<li><a
href="https://github.com/google/flax/commit/aac2f050895f2dde5d96109f106e5af6c425a3ff"><code>aac2f05</code></a>
remove cell_size docs from RNN</li>
<li><a
href="https://github.com/google/flax/commit/0734d0060b6ed9095ccff07300835f59d7376057"><code>0734d00</code></a>
[linen] Only <code>pmean</code> the mean if it is non-zero, in
normalization modules.</li>
<li><a
href="https://github.com/google/flax/commit/1e7a8b1559bf6181829149b1218e980c48eb8791"><code>1e7a8b1</code></a>
[linen] Minor cleanup to normalization code.</li>
<li><a
href="https://github.com/google/flax/commit/1942f3b7f3d9a70984e7fe63b91b407515a1f830"><code>1942f3b</code></a>
Add RMSNorm to documentation</li>
<li><a
href="https://github.com/google/flax/commit/8efa9975700a10a63fd2dbe8c0ec5e4fa7294541"><code>8efa997</code></a>
Add propagation of param_dtype to carry initializer.</li>
<li><a
href="https://github.com/google/flax/commit/b05c673880fbb5e4fc107ecac4ef2921f3e80e2a"><code>b05c673</code></a>
Merge pull request <a
href="https://redirect.github.com/google/flax/issues/3172">#3172</a>
from chiamp:add_absl</li>
<li>Additional commits viewable in <a
href="https://github.com/google/flax/compare/v0.6.5...v0.7.0

">compare
view</a></li>
</ul>
</details>
<br />


Dependabot will resolve any conflicts with this PR as long as you don't
alter it yourself. You can also trigger a rebase manually by commenting
`@dependabot rebase`.

[//]: # (dependabot-automerge-start)
[//]: # (dependabot-automerge-end)

---

<details>
<summary>Dependabot commands and options</summary>
<br />

You can trigger Dependabot actions by commenting on this PR:
- `@dependabot rebase` will rebase this PR
- `@dependabot recreate` will recreate this PR, overwriting any edits
that have been made to it
- `@dependabot merge` will merge this PR after your CI passes on it
- `@dependabot squash and merge` will squash and merge this PR after
your CI passes on it
- `@dependabot cancel merge` will cancel a previously requested merge
and block automerging
- `@dependabot reopen` will reopen this PR if it is closed
- `@dependabot close` will close this PR and stop Dependabot recreating
it. You can achieve the same result by closing it manually
- `@dependabot ignore this major version` will close this PR and stop
Dependabot creating any more for this major version (unless you reopen
the PR or upgrade to it yourself)
- `@dependabot ignore this minor version` will close this PR and stop
Dependabot creating any more for this minor version (unless you reopen
the PR or upgrade to it yourself)
- `@dependabot ignore this dependency` will close this PR and stop
Dependabot creating any more for this dependency (unless you reopen the
PR or upgrade to it yourself)


</details>

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