diff --git a/requirements_jax.txt b/requirements_jax.txt
index 1fe455c3..f07c7383 100644
--- a/requirements_jax.txt
+++ b/requirements_jax.txt
@@ -1 +1 @@
-jax==0.4.20
+jax==0.4.21