diff --git a/setup.py b/setup.py
index 791db95..597886f 100644
--- a/setup.py
+++ b/setup.py
@@ -81,7 +81,7 @@ def parse_requirements(file_path: Path):
             'coremltools>=7.0',
             'openvino-dev>=2023.0',
             'tensorflow<=2.13.1',  # TF bug https://github.com/yolov8nd/yolov8nd/issues/5161
-            'jax<=0.4.21',  # tensorflowjs bug https://github.com/google/jax/issues/18978
+            'jax<=0.4.25',  # tensorflowjs bug https://github.com/google/jax/issues/18978
             'jaxlib<=0.4.21',  # tensorflowjs bug https://github.com/google/jax/issues/18978
             'tensorflowjs',  # automatically installs tensorflow
         ], },