pip prod(deps): bump jax from 0.4.31 to 0.4.33

Bumps [jax](https://github.com/google/jax) from 0.4.31 to 0.4.33.
- [Release notes](https://github.com/google/jax/releases)
- [Changelog](https://github.com/google/jax/blob/main/CHANGELOG.md)
- [Commits](jax-ml/jax@jax-v0.4.31...jax-v0.4.33)

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

Signed-off-by: dependabot[bot] <support@github.com>

pyproject.toml

@@ -34,7 +34,7 @@ dev = [
     "pyright==1.1.345",
 ]
 gui = ["textual==0.79.1", "pyclip==0.7"]
-jax = ["jax==0.4.31", "numpy==2.1.1"]
+jax = ["jax==0.4.33", "numpy==2.1.1"]
 onnx = ["onnx==1.16.2", "numpy==2.1.1"]
 riscv = ["riscemu==2.2.7"]
 wgpu = ["wgpu==0.17.3"]