diff --git a/setup.py b/setup.py
index eb7649a..734cc04 100644
--- a/setup.py
+++ b/setup.py
@@ -11,6 +11,7 @@ INSTALL_REQ += [
     "mpi4py",
     "numba",
     "numpyro",
+    "interpax"
     "quadax",
     "scikit-learn",
     "tqdm",