diff --git a/tests/constraints.txt b/tests/constraints.txt
index b964d71d91509f..eace6a56940a09 100644
--- a/tests/constraints.txt
+++ b/tests/constraints.txt
@@ -23,7 +23,7 @@ pytest-dependency==0.5.1
 pytest-html==4.1.1
 pytest-timeout==2.2.0
 jax<=0.4.14
-jaxlib<=0.4.14
+jaxlib<=0.4.31
 kornia==0.7.0
 networkx<=3.1.0
 keras>=2.0.0,<3.0.0