change jax python version

priyakasimbeg · priyakasimbeg · commit c9b641158e5f · 2025-01-16T23:28:20.000Z
diff --git 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
