diff --git a/gigl-core/pyproject.toml b/gigl-core/pyproject.toml
index 8fc595572..cea3df65f 100644
--- a/gigl-core/pyproject.toml
+++ b/gigl-core/pyproject.toml
@@ -2,7 +2,7 @@
 name            = "gigl-core"
 description     = "GiGL C++/CUDA kernels (pybind11 extensions)"
 readme          = "README.md"
-version         = "0.2.0"
+version         = "0.2.1"
 requires-python = "==3.11.*"
 # Torch is resolved from the ambient environment. gigl-core wheels are ABI-bound
 # to the torch variant they were built against (cpu or cu128). The parent `gigl`
diff --git a/gigl/__init__.py b/gigl/__init__.py
index d3ec452c3..3ced3581b 100644
--- a/gigl/__init__.py
+++ b/gigl/__init__.py
@@ -1 +1 @@
-__version__ = "0.2.0"
+__version__ = "0.2.1"
diff --git a/gigl/dep_vars.env b/gigl/dep_vars.env
index 4b28e38b7..9f1b630dc 100644
--- a/gigl/dep_vars.env
+++ b/gigl/dep_vars.env
@@ -3,11 +3,11 @@ DOCKER_LATEST_BASE_CUDA_IMAGE_NAME_WITH_TAG=us-central1-docker.pkg.dev/external-
 DOCKER_LATEST_BASE_CPU_IMAGE_NAME_WITH_TAG=us-central1-docker.pkg.dev/external-snap-ci-github-gigl/public-gigl/gigl-cpu-base:b598f3d72eee47f5513dcb39460944459a0a012f.108.1
 DOCKER_LATEST_BASE_DATAFLOW_IMAGE_NAME_WITH_TAG=us-central1-docker.pkg.dev/external-snap-ci-github-gigl/public-gigl/gigl-dataflow-base:b598f3d72eee47f5513dcb39460944459a0a012f.108.1
 
-DEFAULT_GIGL_RELEASE_SRC_IMAGE_CUDA=us-central1-docker.pkg.dev/external-snap-ci-github-gigl/public-gigl/src-cuda:0.2.0
-DEFAULT_GIGL_RELEASE_SRC_IMAGE_CPU=us-central1-docker.pkg.dev/external-snap-ci-github-gigl/public-gigl/src-cpu:0.2.0
-DEFAULT_GIGL_RELEASE_SRC_IMAGE_DATAFLOW_CPU=us-central1-docker.pkg.dev/external-snap-ci-github-gigl/public-gigl/src-cpu-dataflow:0.2.0
-DEFAULT_GIGL_RELEASE_DEV_WORKBENCH_IMAGE=us-central1-docker.pkg.dev/external-snap-ci-github-gigl/public-gigl/gigl-dev-workbench:0.2.0
-DEFAULT_GIGL_RELEASE_KFP_PIPELINE_PATH=gs://public-gigl/releases/pipelines/gigl-pipeline-0.2.0.yaml
+DEFAULT_GIGL_RELEASE_SRC_IMAGE_CUDA=us-central1-docker.pkg.dev/external-snap-ci-github-gigl/public-gigl/src-cuda:0.2.1
+DEFAULT_GIGL_RELEASE_SRC_IMAGE_CPU=us-central1-docker.pkg.dev/external-snap-ci-github-gigl/public-gigl/src-cpu:0.2.1
+DEFAULT_GIGL_RELEASE_SRC_IMAGE_DATAFLOW_CPU=us-central1-docker.pkg.dev/external-snap-ci-github-gigl/public-gigl/src-cpu-dataflow:0.2.1
+DEFAULT_GIGL_RELEASE_DEV_WORKBENCH_IMAGE=us-central1-docker.pkg.dev/external-snap-ci-github-gigl/public-gigl/gigl-dev-workbench:0.2.1
+DEFAULT_GIGL_RELEASE_KFP_PIPELINE_PATH=gs://public-gigl/releases/pipelines/gigl-pipeline-0.2.1.yaml
 
 SPARK_31_TFRECORD_JAR_GCS_PATH=gs://public-gigl/tools/scala/spark_packages/spark-custom-tfrecord_2.12-0.5.0.jar
 SPARK_35_TFRECORD_JAR_GCS_PATH=gs://public-gigl/tools/scala/spark_packages/spark_3.5.0-custom-tfrecord_2.12-0.6.1.jar
diff --git a/pyproject.toml b/pyproject.toml
index ebbc65ea8..06cfe5746 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -2,7 +2,7 @@
 name            = "gigl"
 description     = "GIgantic Graph Learning Library"
 readme          = "README.md"
-version         = "0.2.0"
+version         = "0.2.1"
 classifiers     = [
     "Programming Language :: Python",
     "Programming Language :: Python :: 3",
@@ -15,7 +15,7 @@ dependencies = [
   "chardet",
   # gigl-core hosts all C++ / CUDA / pybind11 extensions. Separate wheel per torch
   # variant (cpu/cu128). Version must match gigl exactly.
-  "gigl-core==0.2.0",
+  "gigl-core==0.2.1",
   "google-cloud-aiplatform",
   "google-cloud-dataproc",
   "google-cloud-logging",
diff --git a/uv.lock b/uv.lock
index b8303f693..7bef425e0 100644
--- a/uv.lock
+++ b/uv.lock
@@ -702,7 +702,7 @@ wheels = [
 
 [[package]]
 name = "gigl"
-version = "0.2.0"
+version = "0.2.1"
 source = { editable = "." }
 dependencies = [
     { name = "argo-workflows" },
@@ -963,7 +963,7 @@ typing-stubs = [
 
 [[package]]
 name = "gigl-core"
-version = "0.2.0"
+version = "0.2.1"
 source = { directory = "gigl-core" }
 
 [[package]]