pip prod(deps): bump jax from 0.4.36 to 0.4.37 (#3612)

Bumps [jax](https://github.com/jax-ml/jax) from 0.4.36 to 0.4.37.
<details>
<summary>Release notes</summary>
<p><em>Sourced from <a
href="https://github.com/jax-ml/jax/releases">jax's
releases</a>.</em></p>
<blockquote>
<h2>JAX v0.4.37</h2>
<p>This is a patch release of jax 0.4.36. Only &quot;jax&quot; was
released at this version.</p>
<ul>
<li>Bug fixes
<ul>
<li>Fixed a bug where <code>jit</code> would error if an argument was
named <code>f</code> (<a
href="https://redirect.github.com/jax-ml/jax/issues/25329">#25329</a>).</li>
<li>Fix a bug that will throw <code>index out of range</code> error in
<code>jax.lax.while_loop</code> if the user registers pytree node class
with
different aux data for the flatten and flatten_with_path.</li>
<li>Pinned a new libtpu release (0.0.6) that fixes a compiler bug on TPU
v6e.</li>
</ul>
</li>
</ul>
</blockquote>
</details>
<details>
<summary>Changelog</summary>
<p><em>Sourced from <a
href="https://github.com/jax-ml/jax/blob/main/CHANGELOG.md">jax's
changelog</a>.</em></p>
<blockquote>
<h2>jax 0.4.37 (Dec 9, 2024)</h2>
<p>This is a patch release of jax 0.4.36. Only &quot;jax&quot; was
released at this version.</p>
<ul>
<li>Bug fixes
<ul>
<li>Fixed a bug where <code>jit</code> would error if an argument was
named <code>f</code> (<a
href="https://redirect.github.com/jax-ml/jax/issues/25329">#25329</a>).</li>
<li>Fix a bug that will throw <code>index out of range</code> error in
{func}<code>jax.lax.while_loop</code> if the user register pytree node
class with
different aux data for the flatten and flatten_with_path.</li>
<li>Pinned a new libtpu release (0.0.6) that fixes a compiler bug on TPU
v6e.</li>
</ul>
</li>
</ul>
</blockquote>
</details>
<details>
<summary>Commits</summary>
<ul>
<li><a
href="https://github.com/jax-ml/jax/commit/ffb07cdadb5dc3bc43485cf041dbc2b43136109e"><code>ffb07cd</code></a>
Update versions for v0.4.37 release.</li>
<li><a
href="https://github.com/jax-ml/jax/commit/95892fdac86524151b6dadd7d8bedbf915f1500f"><code>95892fd</code></a>
Use private names for args in api_util to avoid shadowing kwargs
keys.</li>
<li><a
href="https://github.com/jax-ml/jax/commit/65b60884114261549ffc2eb937162bdeaa493928"><code>65b6088</code></a>
Avoid index out of range error in carry structure check</li>
<li>See full diff in <a
href="https://github.com/jax-ml/jax/compare/jax-v0.4.36...jax-v0.4.37">compare
view</a></li>
</ul>
</details>
<br />


[![Dependabot compatibility
score](https://dependabot-badges.githubapp.com/badges/compatibility_score?dependency-name=jax&package-manager=pip&previous-version=0.4.36&new-version=0.4.37)](https://docs.github.com/en/github/managing-security-vulnerabilities/about-dependabot-security-updates#about-compatibility-scores)

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 show <dependency name> ignore conditions` will show all
of the ignore conditions of the specified dependency
- `@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] <[email protected]>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

pyproject.toml

@@ -34,7 +34,7 @@ dev = [
     "pyright==1.1.390",
 ]
 gui = ["textual==0.89.1", "pyclip==0.7"]
-jax = ["jax==0.4.36", "numpy==2.2.0"]
+jax = ["jax==0.4.37", "numpy==2.2.0"]
 onnx = ["onnx==1.17.0", "numpy==2.2.0"]
 riscv = ["riscemu==2.2.7"]
 wgpu = ["wgpu==0.19.2"]