Remove obsolete comment

PiperOrigin-RevId: 556843918
google · Aug 14, 2023 · 3409728 · 3409728
1 parent e50c0bc
commit 3409728
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/notebooks/weight_space_linearization.ipynb b/notebooks/weight_space_linearization.ipynb
@@ -63,9 +63,7 @@
     "pycharm": {}
    },
    "source": [
-    "Install JAX, Tensorflow Datasets, and Neural Tangents\n",
-    "\n",
-    "The first line specifies the version of jaxlib that we would like to import. Note, that \"cp36\" species the version of python (version 3.6) used by JAX. Make sure your colab kernel matches this version."
+    "Install JAX, Tensorflow Datasets, and Neural Tangents"
    ]
   },
   {
@@ -79,10 +77,10 @@
    },
    "outputs": [],
    "source": [
-        "!pip install -q --upgrade pip\n",
-        "!pip install -q --upgrade jax[cuda11_cudnn82] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html\n",
-        "!pip install -q tensorflow-datasets\n",
-        "!pip install -q git+https://www.github.com/google/neural-tangents"
+    "!pip install -q --upgrade pip\n",
+    "!pip install -q --upgrade jax[cuda11_cudnn82] -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html\n",
+    "!pip install -q tensorflow-datasets\n",
+    "!pip install -q git+https://www.github.com/google/neural-tangents"
    ]
   },
   {