diff --git a/pyproject.toml b/pyproject.toml
index 0dca8cf..61daf2b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -19,7 +19,7 @@ dependencies = [
     "agjax",
     "ceviche_challenges",
     "fmmax",
-    "jax <= 0.4.37",
+    "jax <= 0.4.38",
     "jaxlib",
     "numpy",
     "refractiveindex",