diff --git a/requirements_jax.txt b/requirements_jax.txt
index 626c0f05..cd85e1ca 100644
--- a/requirements_jax.txt
+++ b/requirements_jax.txt
@@ -1 +1 @@
-jax==0.4.27
+jax==0.4.28