diff --git a/tests/constraints.txt b/tests/constraints.txt
index 671c60bc937c83..b4d2ff2cfe2d9d 100644
--- a/tests/constraints.txt
+++ b/tests/constraints.txt
@@ -21,5 +21,5 @@ pytest-dependency==0.5.1
 pytest-html==3.2.0
 pytest-timeout==2.1.0
 jax<=0.4.14
-jaxlib<=0.4.14
+jaxlib<=0.4.18
 torch<2.1.0
\ No newline at end of file