diff --git a/setup.cfg b/setup.cfg
index f926a3e5..b5bc2e59 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -39,7 +39,7 @@ dev =
     yapf
 
 jaxcpu =
-    jax==0.8.0
+    jax==0.8.2
 
 [options.entry_points]
 console_scripts =