From 33f826aaf5b9604f181dd0a03428f9a33dfd6c33 Mon Sep 17 00:00:00 2001
From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com>
Date: Tue, 10 Dec 2024 03:33:35 +0000
Subject: [PATCH] Bump jax from 0.4.36 to 0.4.37 (#977)

Bumps [jax](https://github.com/jax-ml/jax) from 0.4.36 to 0.4.37.
- [Release notes](https://github.com/jax-ml/jax/releases)
- [Changelog](https://github.com/jax-ml/jax/blob/main/CHANGELOG.md)
- [Commits](https://github.com/jax-ml/jax/compare/jax-v0.4.36...jax-v0.4.37)

---
updated-dependencies:
- dependency-name: jax
  dependency-type: direct:development
  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>
---
 poetry.lock | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/poetry.lock b/poetry.lock
index 79115641..aed38e44 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -560,17 +560,17 @@ files = [
 
 [[package]]
 name = "jax"
-version = "0.4.36"
+version = "0.4.37"
 description = "Differentiate, compile, and transform Numpy code."
 optional = false
 python-versions = ">=3.10"
 files = [
-    {file = "jax-0.4.36-py3-none-any.whl", hash = "sha256:4d86a99fab7c25ea0deda2672b429ef650e5eac819884e74d35c32c6e86a5856"},
-    {file = "jax-0.4.36.tar.gz", hash = "sha256:088bff0575d01fc82682a9af4eb07433d60de7e5164686bd2cea3439492e608a"},
+    {file = "jax-0.4.37-py3-none-any.whl", hash = "sha256:bdc0686d7e5a944e2d38026eae632214d98dd2d91869cbcedbf1c11298ae3e3e"},
+    {file = "jax-0.4.37.tar.gz", hash = "sha256:7774f3d9e23fe199c65589c680c5a5be87a183b89598421a632d8245222b637b"},
 ]
 
 [package.dependencies]
-jaxlib = "0.4.36"
+jaxlib = ">=0.4.36,<=0.4.37"
 ml_dtypes = ">=0.4.0"
 numpy = [
     {version = ">=1.26.0", markers = "python_version >= \"3.12\""},
@@ -584,13 +584,13 @@ scipy = [
 
 [package.extras]
 ci = ["jaxlib (==0.4.35)"]
-cuda = ["jax-cuda12-plugin[with-cuda] (==0.4.36)", "jaxlib (==0.4.36)"]
-cuda12 = ["jax-cuda12-plugin[with-cuda] (==0.4.36)", "jaxlib (==0.4.36)"]
+cuda = ["jax-cuda12-plugin[with-cuda] (>=0.4.36,<=0.4.37)", "jaxlib (==0.4.36)"]
+cuda12 = ["jax-cuda12-plugin[with-cuda] (>=0.4.36,<=0.4.37)", "jaxlib (==0.4.36)"]
 cuda12-local = ["jax-cuda12-plugin (==0.4.36)", "jaxlib (==0.4.36)"]
-cuda12-pip = ["jax-cuda12-plugin[with-cuda] (==0.4.36)", "jaxlib (==0.4.36)"]
+cuda12-pip = ["jax-cuda12-plugin[with-cuda] (>=0.4.36,<=0.4.37)", "jaxlib (==0.4.36)"]
 k8s = ["kubernetes"]
 minimum-jaxlib = ["jaxlib (==0.4.36)"]
-tpu = ["jaxlib (==0.4.36)", "libtpu (==0.0.5)", "libtpu-nightly (==0.1.dev20241010+nightly.cleanup)", "requests"]
+tpu = ["jaxlib (>=0.4.36,<=0.4.37)", "libtpu (==0.0.6)", "libtpu-nightly (==0.1.dev20241010+nightly.cleanup)", "requests"]
 
 [[package]]
 name = "jaxlib"