diff --git a/docs/notebooks/quickstart.ipynb b/docs/notebooks/quickstart.ipynb
index d35eb802d..bdaf8fb83 100644
--- a/docs/notebooks/quickstart.ipynb
+++ b/docs/notebooks/quickstart.ipynb
@@ -43,20 +43,6 @@
     "from jax import random"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {
-    "tags": [
-     "remove-cell"
-    ]
-   },
-   "outputs": [],
-   "source": [
-    "# Prevent GPU/TPU warning.\n",
-    "import jax; jax.config.update('jax_platform_name', 'cpu')"
-   ]
-  },
   {
    "cell_type": "markdown",
    "metadata": {
diff --git a/docs/notebooks/quickstart.md b/docs/notebooks/quickstart.md
index 8fc29344b..bd1503767 100644
--- a/docs/notebooks/quickstart.md
+++ b/docs/notebooks/quickstart.md
@@ -45,13 +45,6 @@ from jax import grad, jit, vmap
 from jax import random
 ```
 
-```{code-cell} ipython3
-:tags: [remove-cell]
-
-# Prevent GPU/TPU warning.
-import jax; jax.config.update('jax_platform_name', 'cpu')
-```
-
 +++ {"id": "FQ89jHCYfhpg"}
 
 ## Multiplying Matrices