Minor updates.

Author: ericmjl

environment.yml

@@ -6,8 +6,8 @@ dependencies:
   - black>=20.8b1
   - conda
   - ipywidgets>=7.6.3
-  - jax>=0.2.8
-  - jaxlib>=0.1.59
+  - jax<=0.2.9
+  - jaxlib<=0.1.59
   - jupyter
   - jupyterlab>=3.0.5
   - mamba

notebooks/01-differential-programming/01-neural-nets-from-scratch.ipynb

@@ -290,7 +290,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.9"
+   "version": "3.9.6"
   },
   "toc-autonumbering": true,
   "toc-showcode": false,

notebooks/01-differential-programming/02-gradient-optimization.ipynb

@@ -23,7 +23,9 @@
   },
   {
    "cell_type": "markdown",
-   "metadata": {},
+   "metadata": {
+    "tags": []
+   },
    "source": [
     "## Derivatives\n",
     "\n",
@@ -64,7 +66,7 @@
     "\n",
     "Hence, \n",
     "\n",
-    "$$w = -\\frac{3}{2} = 1.5$$\n",
+    "$$w = -\\frac{3}{2}$$\n",
     "\n",
     "To check whether the value of $w$ at the place where $f'(w) = 0$ is a minima or maxima,\n",
     "we can use another piece of knowledge from 1st year undergraduate calculus:\n",
@@ -267,7 +269,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.2"
+   "version": "3.9.6"
   }
  },
  "nbformat": 4,