add optax installation to docs

ASEM000 · Sep 8, 2023 · 7dd5211 · 7dd5211
1 parent 57035dd
commit 7dd5211
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/docs/notebooks/train_bilstm.ipynb b/docs/notebooks/train_bilstm.ipynb
@@ -13,7 +13,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install git+https://github.com/ASEM000/serket --quiet"
+    "!pip install git+https://github.com/ASEM000/serket --quiet\n",
+    "!pip install optax --quiet"
    ]
   },
   {

diff --git a/docs/notebooks/train_mnist.ipynb b/docs/notebooks/train_mnist.ipynb
@@ -31,7 +31,8 @@
       "outputs": [],
       "source": [
         "!pip install keras-core --quiet\n",
-        "!pip install git+https://github.com/ASEM000/serket --quiet"
+        "!pip install git+https://github.com/ASEM000/serket --quiet\n",
+        "!pip install optax --quiet"
       ]
     },
     {

diff --git a/docs/notebooks/train_mp.ipynb b/docs/notebooks/train_mp.ipynb
@@ -15,7 +15,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip install git+https://github.com/ASEM000/serket --quiet"
+    "!pip install git+https://github.com/ASEM000/serket --quiet\n",
+    "!pip install git+https://github.com/deepmind/jmp --quiet"
    ]
   },
   {