Update mental_model.ipynb

ASEM000 · Aug 17, 2023 · a6a267e · a6a267e
1 parent ec07f15
commit a6a267e
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/docs/notebooks/mental_model.ipynb b/docs/notebooks/mental_model.ipynb
@@ -11,7 +11,6 @@
     "\n",
     "3) **Transformation Transparent Code**: Write code that works under any `jax` transformation. This approach makes the code reusable, easier to understand, and less error-prone.\n",
     "\n",
-    "\n",
     "The following code showcases these principles"
    ]
   },
@@ -40,8 +39,6 @@
     "# to showcase the functional update of with stateful modules\n",
     "# lets define a tree of modules with stateful updates in the call method\n",
     "\n",
-    "\n",
-    "@sk.autoinit\n",
     "class Linears(sk.TreeClass):\n",
     "    def __init__(self):\n",
     "        self.l1 = Linear(1, 120, key=jax.random.PRNGKey(0))\n",