Add GPU support

parsiad · Aug 29, 2024 · 389e5f7 · 389e5f7
1 parent 8c63920
commit 389e5f7
Show file tree

Hide file tree

Showing 15 changed files with 308 additions and 161 deletions.
diff --git a/examples/mnist.ipynb b/examples/mnist.ipynb
diff --git a/examples/n-gram.ipynb b/examples/n-gram.ipynb
@@ -39,8 +39,9 @@
    "outputs": [],
    "source": [
     "import micrograd_pp as mpp\n",
-    "import numpy as np\n",
-    "import numpy.typing as npt"
+    "import numpy.typing as npt\n",
+    "\n",
+    "np = mpp.numpy  # `import numpy as np` will not work when using GPU"
    ]
   },
   {
@@ -324,7 +325,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.12.4"
+   "version": "3.12.5"
   }
  },
  "nbformat": 4,