From d563f837f396b7d1bea933539eb3163ae871b5c9 Mon Sep 17 00:00:00 2001
From: nstarman <nstarman@users.noreply.github.com>
Date: Wed, 11 Dec 2024 11:30:55 -0500
Subject: [PATCH] bump: set jax v0.4.20+

Signed-off-by: nstarman <nstarman@users.noreply.github.com>
---
 pyproject.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 5657e3d..c814189 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -23,8 +23,8 @@
     "Typing :: Typed",
   ]
   dependencies = [
-    "jax>0.4",
-    "jaxlib>0.4",
+    "jax>0.4.20",
+    "jaxlib>0.4.20",
     "jaxtyping>=0.2.34",
     "plum-dispatch>=2.5.2",
     "quax>0.0.4",