Updates to Python interface for PyPI packaging (#1209)

* Updates

* Updates to notebooks

examples/python/02_pytorch_extension_grouped_gemm.ipynb

@@ -10,6 +10,52 @@
     "This notebook walks through a basic example of using the CUTLASS Python interface to declare\n",
     "a grouped GEMM kernel and export it as a PyTorch CUDA extension. Note that GEMM and Conv2d can also be exported as PyTorch CUDA extensions. \n",
     "\n",
+    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/NVIDIA/cutlass/tree/master/examples/python/02_pytorch_extension_grouped_gemm.ipynb)\n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "2d70560e",
+   "metadata": {},
+   "source": [
+    "## Prerequisites for running on Colab\n",
+    "This notebook requires an NVIDIA GPU. If `nvidia-smi` fails, go to Runtime -> Change runtime type -> Hardware accelerator and confirm a GPU is selected."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "cc7c7458",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!#nvidia-smi"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "2107bb0d",
+   "metadata": {},
+   "source": [
+    "If running on Colab, you will need to install the CUTLASS Python interface and PyTorch. To do so, uncomment the following line and run the cell:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "a9852cb8",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!#pip install nvidia-cutlass torch --extra-index-url https://download.pytorch.org/whl/cu121"
+   ]
+  },
+  {
+   "attachments": {},
+   "cell_type": "markdown",
+   "id": "962324fd",
+   "metadata": {},
+   "source": [
     "## Background on grouped GEMM\n",
     "Grouped GEMM enables one to execute a set of GEMMs (each with potentially different sizes and strides)\n",
     "in a single CUDA kernel. It can be thought of as a generalized version of a pointer-array GEMM,\n",