Update [recipes]sharing.ipynb

docs/notebooks/[recipes]sharing.ipynb

@@ -50,7 +50,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "In this example a simple `AutoEncoder` with shared `weight` between the encode/decoder is demonstrated. "
+    "In this example a simple `AutoEncoder` with shared `weight` between the encode/decoder is demonstrated."
    ]
   },
   {
@@ -66,7 +66,9 @@
     "import functools as ft\n",
     "\n",
     "\n",
-    "def stateful(method):\n",
+    "def sharing(method):\n",
+    "    # sharing simply copies the instance, executes the method, and returns the output\n",
+    "    # **without modifying the original instance.**\n",
     "    @ft.wraps(method)\n",
     "    def wrapper(self, *args, **kwargs):\n",
     "        # `value_and_tree` executes any mutating method in a functional way\n",
@@ -95,7 +97,7 @@
     "        self.dec2 = Linear(20, 10, key=k3)\n",
     "        self.dec1 = Linear(10, 1, key=k4)\n",
     "\n",
-    "    @stateful\n",
+    "    @sharing\n",
     "    def tied_call(self, input: jax.Array) -> jax.Array:\n",
     "        self.dec1.weight = self.enc1.weight.T\n",
     "        self.dec2.weight = self.enc2.weight.T\n",