Update extras_require in setup.py

asmeurer · asmeurer · commit dfac54062c57 · 2024-03-22T16:24:31.000-06:00
diff --git a/setup.py b/setup.py
@@ -18,6 +18,9 @@
     extras_require={
         "numpy": "numpy",
         "cupy": "cupy",
+        "jax": "jax",
+        "pytorch": "pytorch",
+        "dask": "dask",
     },
     classifiers=[
         "Programming Language :: Python :: 3",
