diff --git a/tests/constraints.txt b/tests/constraints.txt
index 616aea79c82153..4853e7ba9a0d9d 100644
--- a/tests/constraints.txt
+++ b/tests/constraints.txt
@@ -21,7 +21,7 @@ pytest>=5.0,<8.4
 pytest-dependency==0.5.1
 pytest-html==4.1.1
 pytest-timeout==2.3.1
-jax<=0.4.33
+jax<=0.4.34
 jaxlib<=0.4.33
 kornia==0.7.0
 networkx<=3.3