diff --git a/setup.py b/setup.py
index c8d5218eb..43652b01c 100644
--- a/setup.py
+++ b/setup.py
@@ -22,7 +22,7 @@ def readme():
         "tf-keras>=2.16.0, <2.19.0",
     ],
     "keops": [
-        "pykeops>=2.0.0, <2.3.0",
+        "pykeops>=2.0.0, <2.4.0",
         "torch>=1.7.0, <3.0.0"
     ],
     "all": [
@@ -30,7 +30,7 @@ def readme():
         "tensorflow_probability>=0.8.0, <0.26.0",
         "tensorflow>=2.16.0, <2.19.0",
         "tf-keras>=2.16.0, <2.19.0",
-        "pykeops>=2.0.0, <2.3.0",
+        "pykeops>=2.0.0, <2.4.0",
         "torch>=1.7.0, <3.0.0"
     ],
 }