diff --git a/tests/constraints.txt b/tests/constraints.txt
index 16bffdf16967db..bab775ec94eef2 100644
--- a/tests/constraints.txt
+++ b/tests/constraints.txt
@@ -22,7 +22,7 @@ pytest>=5.0,<8.4
 pytest-dependency==0.5.1
 pytest-html==4.1.1
 pytest-timeout==2.2.0
-jax<=0.4.14
+jax<=0.4.35
 jaxlib<=0.4.14
 kornia==0.7.0
 networkx<=3.3