chore: remove jaxlib cpu and gpu deps

mthiboust · Nov 23, 2023 · 6346103 · 6346103
1 parent bc582c7
commit 6346103
Showing 1 changed file with 0 additions and 15 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -31,21 +31,6 @@ notebook = [
     "ipywidgets>=8.1.1",
     "python-dotenv>=1.0.0",
 ]
-cpu = [
-    "jaxlib==0.4.20",
-]
-gpu = [
-    "jaxlib @ https://storage.googleapis.com/jax-releases/cuda12/jaxlib-0.4.20+cuda12.cudnn89-cp311-cp311-manylinux2014_x86_64.whl",
-    "nvidia-cublas-cu12>=12.2.5.6",
-    "nvidia-cuda-cupti-cu12>=12.2.142",
-    "nvidia-cuda-nvcc-cu12>=12.2.140",
-    "nvidia-cuda-runtime-cu12>=12.2.140",
-    "nvidia-cudnn-cu12>=8.9",
-    "nvidia-cufft-cu12>=11.0.8.103",
-    "nvidia-cusolver-cu12>=11.5.2",
-    "nvidia-cusparse-cu12>=12.1.2.141",
-    "nvidia-nccl-cu12>=2.18.3",
-]
 dev = [
     "pytest>=7.4.3",
     "ruff>=0.1.5",