diff --git a/setup.py b/setup.py
index 5a3bb4c1d1981fc9d583036162c190aae72b997e..ab666ecc4049db86143714ecc4fe27a681b12c37 100644
--- a/setup.py
+++ b/setup.py
@@ -146,7 +146,7 @@ setup(
     zip_safe=True,
     dependency_links=[],
     install_requires=["ducc0>=0.23.0", "numpy", "nifty8>=8.0"],
-    extras_require={"full": ("astropy", "pytest", "pytest-cov", "mpi4py", "python-casacore", "h5py", "matplotlib")},
+    extras_require={"full": ("astropy", "pytest", "pytest-cov", "mpi4py", "python-casacore", "h5py", "matplotlib", "jax", "jaxlib", "jaxbind")},
     ext_modules=extensions,
     entry_points={"console_scripts":
         [