diff --git a/MODULE.bazel b/MODULE.bazel
index f5fefee6b..687f96355 100644
--- a/MODULE.bazel
+++ b/MODULE.bazel
@@ -4,7 +4,7 @@ pyrovelocity MODULE
 
 module(
     name = "pyrovelocity",
-    version = "0.2.0b22",
+    version = "0.2.0",
     compatibility_level = 1,
 )
 
diff --git a/Makefile b/Makefile
index 0f4fa90b2..cd48fea6a 100644
--- a/Makefile
+++ b/Makefile
@@ -913,8 +913,8 @@ approve-prs: ## Approve github pull requests from bots: PR_ENTRIES="2-5 10 12-18
 		fi; \
 	done
 
-PREVIOUS_VERSION := 0.2.0b21
-NEXT_VERSION := 0.2.0b22
+PREVIOUS_VERSION := 0.2.0b22
+NEXT_VERSION := 0.2.0
 VERSION_FILES := \
 	pyproject.toml \
 	conda/colab/construct.yaml \
diff --git a/conda/colab/construct.yaml b/conda/colab/construct.yaml
index ef42de36e..6746d8e35 100644
--- a/conda/colab/construct.yaml
+++ b/conda/colab/construct.yaml
@@ -1,5 +1,5 @@
 name: pyrovelocity-colab
-version: 0.2.0b22
+version: 0.2.0
 
 channels:
   - pytorch
diff --git a/containers/gpu.Dockerfile b/containers/gpu.Dockerfile
index a95841020..b42d4b8e5 100644
--- a/containers/gpu.Dockerfile
+++ b/containers/gpu.Dockerfile
@@ -77,7 +77,7 @@ COPY . /root
 # development
 RUN pip install --no-deps -e .
 # distribution
-# RUN pip install pyrovelocity==0.2.0b22
+# RUN pip install pyrovelocity==0.2.0
 
 ARG tag
 ENV FLYTE_INTERNAL_IMAGE $tag
diff --git a/containers/pkg.Dockerfile b/containers/pkg.Dockerfile
index 6c27b5945..f09985652 100644
--- a/containers/pkg.Dockerfile
+++ b/containers/pkg.Dockerfile
@@ -34,7 +34,7 @@ COPY . /root
 # development
 RUN pip install --no-deps -e .
 # distribution
-# RUN pip install pyrovelocity==0.2.0b22
+# RUN pip install pyrovelocity==0.2.0
 
 ARG tag
 ENV FLYTE_INTERNAL_IMAGE $tag
diff --git a/docs/source/notebooks/pyrovelocity_colab_template.ipynb b/docs/source/notebooks/pyrovelocity_colab_template.ipynb
index d22ea7539..249e7356f 100644
--- a/docs/source/notebooks/pyrovelocity_colab_template.ipynb
+++ b/docs/source/notebooks/pyrovelocity_colab_template.ipynb
@@ -70,10 +70,11 @@
         }
       ],
       "source": [
-        "pyrovelocity_version = \"0.2.0b22\"\n",
-        "pyrovelocity_colab_script_url = \\\n",
-        "\"https://storage.googleapis.com/pyrovelocity/data/scripts/\" + \\\n",
-        "f\"pyrovelocity-colab-{pyrovelocity_version}-Linux-x86_64.sh\"\n",
+        "pyrovelocity_version = \"0.2.0\"\n",
+        "pyrovelocity_colab_script_url = (\n",
+        "    \"https://storage.googleapis.com/pyrovelocity/data/scripts/\"\n",
+        "    + f\"pyrovelocity-colab-{pyrovelocity_version}-Linux-x86_64.sh\"\n",
+        ")\n",
         "\n",
         "!pip install -q condacolab\n",
         "import condacolab\n",
@@ -116,7 +117,7 @@
         "\n",
         "# change the branch argument from a tag to a branch in development\n",
         "git clone \\\n",
-        "--branch beta \\\n",
+        "--branch main \\\n",
         "https://github.com/pinellolab/pyrovelocity.git"
       ]
     },
@@ -141,6 +142,7 @@
       "source": [
         "!pip install --user --no-deps -e pyrovelocity\n",
         "from IPython import get_ipython\n",
+        "\n",
         "get_ipython().kernel.do_shutdown(True)"
       ]
     },
@@ -173,6 +175,7 @@
       "outputs": [],
       "source": [
         "import pyrovelocity\n",
+        "\n",
         "print(pyrovelocity.__file__)\n",
         "print(pyrovelocity.__version__)"
       ]
diff --git a/pyproject.toml b/pyproject.toml
index 67172f5f4..74eef80c3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyrovelocity"
-version = "0.2.0b22"
+version = "0.2.0"
 packages = [{ include = "pyrovelocity", from = "src" }]
 description = "A multivariate RNA Velocity model to estimate future cell states with uncertainty using probabilistic modeling with pyro."
 authors = ["pyrovelocity team"]
diff --git a/scripts/conda b/scripts/conda
index 40da38ea7..b57a50572 100755
--- a/scripts/conda
+++ b/scripts/conda
@@ -3,10 +3,11 @@
 set -euo pipefail
 
 PACKAGE_NAME="pyrovelocity"
-PACKAGE_VERSION="0.2.0b22"
+PACKAGE_VERSION="0.2.0"
 CONDA_BUILD_STRING="pyhff70e4c"
 CONDA_BUILD_NUMBER="0"
-CONDA_CHANNEL_LABEL="pyrovelocity_dev"
+# CONDA_CHANNEL_LABEL="pyrovelocity_dev"
+CONDA_CHANNEL_LABEL="main"
 VERBOSE=0
 USE_COLOR=true
 
@@ -31,7 +32,7 @@ Example:
 
     ./conda \\
     --name pyrovelocity \\
-    --version 0.2.0b22.dev1 \\
+    --version 0.2.0.dev1 \\
     --build-string pyhff70e4c \\
     --build-number 0 \\
     --label pyrovelocity_dev