diff --git a/pyproject.toml b/pyproject.toml
index 747581c8..8eb2c04a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,6 +20,8 @@ classifiers = [
 requires-python = ">= 3.10"
 
 dependencies = [
+    "jax",
+    "jaxlib",
     "apsw",
     "bbknn",
     "cgatcore",