diff --git a/setup.cfg b/setup.cfg
index 040d1e26a..8c512d32e 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -121,8 +121,8 @@ jax_core_deps =
 
 # JAX CPU
 jax_cpu =
-  jax==0.4.38
-  jaxlib==0.4.38
+  jax==0.4.36
+  jaxlib==0.4.36
   %(jax_core_deps)s
 
 # JAX GPU