setup: add jax and jaxbind for full install

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":
         [