diff --git a/setup.cfg b/setup.cfg
index 5d630d7..a389cc9 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -65,7 +65,7 @@ catboost =
 GBDT =
     lightgbm < 4
 neural_tangents =
-    jaxlib <= 0.1.65
+    jaxlib <= 0.5.1
     jax <= 0.2.12
     neural_tangents < 0.6.0
 gpflow =
@@ -79,7 +79,7 @@ all =
     GPy < 1.10
     matplotlib
     lightgbm < 4
-    jaxlib <= 0.1.65
+    jaxlib <= 0.5.1
     jax <= 0.2.12
     neural_tangents < 0.6.0
     tensorflow >= 2.5, < 2.9