diff --git a/tests/constraints.txt b/tests/constraints.txt
index 5ecacb4b65c03e..bf6287bb8b2cbe 100644
--- a/tests/constraints.txt
+++ b/tests/constraints.txt
@@ -23,5 +23,5 @@ pytest-dependency==0.5.1
 pytest-html==4.1.1
 pytest-timeout==2.1.0
 jax<=0.4.14
-jaxlib<=0.4.14
+jaxlib<=0.4.24
 torch>=1.13,<2.2