From 67ef77e91b50b3324e70045215f8d010b8ce6ff1 Mon Sep 17 00:00:00 2001
From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com>
Date: Mon, 20 Jan 2025 07:34:25 -0800
Subject: [PATCH] Bump jax from 0.4.38 to 0.5.0 (#153)

Bumps [jax](https://github.com/jax-ml/jax) from 0.4.38 to 0.5.0.
- [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.38...jax-v0.5.0)

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

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 9e6c61c..c990839 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,7 +16,7 @@ keywords = []
 # pip dependencies of the project
 dependencies = [
     "chex==0.1.88",
-    "jax==0.4.38",
+    "jax==0.5.0",
     "flax==0.10.2",
     "ml_dtypes==0.4.0",
     "optax==0.2.4",