diff --git a/dataflow/gemma-flex-template/requirements-test.txt b/dataflow/gemma-flex-template/requirements-test.txt
index 6138ec5ac4..5b8308a445 100644
--- a/dataflow/gemma-flex-template/requirements-test.txt
+++ b/dataflow/gemma-flex-template/requirements-test.txt
@@ -1,4 +1,4 @@
-google-cloud-aiplatform==1.158.0
+google-cloud-aiplatform==2.0.0
 google-cloud-dataflow-client==0.8.14
 google-cloud-pubsub==2.28.0
 google-cloud-storage==2.18.2
diff --git a/dataflow/run-inference/requirements-test.txt b/dataflow/run-inference/requirements-test.txt
index a88ea43490..29298701dd 100644
--- a/dataflow/run-inference/requirements-test.txt
+++ b/dataflow/run-inference/requirements-test.txt
@@ -1,4 +1,4 @@
-google-cloud-aiplatform==1.158.0
+google-cloud-aiplatform==2.0.0
 google-cloud-dataflow-client==0.8.14
 google-cloud-storage==2.10.0
 pytest==9.0.3; python_version >= "3.10"
diff --git a/gemma2/requirements.txt b/gemma2/requirements.txt
index 67a5814ea6..838ef67c75 100644
--- a/gemma2/requirements.txt
+++ b/gemma2/requirements.txt
@@ -1,2 +1,2 @@
-google-cloud-aiplatform==1.158.0
+google-cloud-aiplatform==2.0.0
 protobuf==6.33.6
diff --git a/generative_ai/embeddings/requirements.txt b/generative_ai/embeddings/requirements.txt
index 254dc4eef6..76beb17c0a 100644
--- a/generative_ai/embeddings/requirements.txt
+++ b/generative_ai/embeddings/requirements.txt
@@ -2,7 +2,7 @@ pandas==2.2.3; python_version == '3.7'
 pandas==2.2.3; python_version == '3.8'
 pandas==2.2.3; python_version > '3.8'
 pillow==12.2.0
-google-cloud-aiplatform[full]==1.157.0
+google-cloud-aiplatform[full]==2.0.0
 sentencepiece==0.2.0
 google-auth==2.29.0
 anthropic[vertex]==0.28.0
diff --git a/generative_ai/evaluation/requirements.txt b/generative_ai/evaluation/requirements.txt
index 2bcc3b8bdc..4d46a9c26a 100644
--- a/generative_ai/evaluation/requirements.txt
+++ b/generative_ai/evaluation/requirements.txt
@@ -2,7 +2,7 @@ pandas==2.2.3; python_version == '3.7'
 pandas==2.2.3; python_version == '3.8'
 pandas==2.2.3; python_version > '3.8'
 pillow==12.2.0
-google-cloud-aiplatform[full]==1.157.0
+google-cloud-aiplatform[full]==2.0.0
 sentencepiece==0.2.0
 google-auth==2.38.0
 anthropic[vertex]==0.28.0
diff --git a/generative_ai/extensions/requirements.txt b/generative_ai/extensions/requirements.txt
index 2bcc3b8bdc..4d46a9c26a 100644
--- a/generative_ai/extensions/requirements.txt
+++ b/generative_ai/extensions/requirements.txt
@@ -2,7 +2,7 @@ pandas==2.2.3; python_version == '3.7'
 pandas==2.2.3; python_version == '3.8'
 pandas==2.2.3; python_version > '3.8'
 pillow==12.2.0
-google-cloud-aiplatform[full]==1.157.0
+google-cloud-aiplatform[full]==2.0.0
 sentencepiece==0.2.0
 google-auth==2.38.0
 anthropic[vertex]==0.28.0
diff --git a/generative_ai/function_calling/requirements.txt b/generative_ai/function_calling/requirements.txt
index 567c5d05c7..c68cc73f0f 100644
--- a/generative_ai/function_calling/requirements.txt
+++ b/generative_ai/function_calling/requirements.txt
@@ -1,3 +1,3 @@
 google-auth==2.38.0
 openai==1.68.2
-google-cloud-aiplatform==1.157.0
+google-cloud-aiplatform==2.0.0
diff --git a/generative_ai/image_generation/requirements.txt b/generative_ai/image_generation/requirements.txt
index 2bcc3b8bdc..4d46a9c26a 100644
--- a/generative_ai/image_generation/requirements.txt
+++ b/generative_ai/image_generation/requirements.txt
@@ -2,7 +2,7 @@ pandas==2.2.3; python_version == '3.7'
 pandas==2.2.3; python_version == '3.8'
 pandas==2.2.3; python_version > '3.8'
 pillow==12.2.0
-google-cloud-aiplatform[full]==1.157.0
+google-cloud-aiplatform[full]==2.0.0
 sentencepiece==0.2.0
 google-auth==2.38.0
 anthropic[vertex]==0.28.0
diff --git a/generative_ai/labels/requirements.txt b/generative_ai/labels/requirements.txt
index 567a82bafe..61d1ddf6ff 100644
--- a/generative_ai/labels/requirements.txt
+++ b/generative_ai/labels/requirements.txt
@@ -1 +1 @@
-google-cloud-aiplatform==1.157.0
+google-cloud-aiplatform==2.0.0
diff --git a/generative_ai/model_garden/requirements.txt b/generative_ai/model_garden/requirements.txt
index 2bcc3b8bdc..4d46a9c26a 100644
--- a/generative_ai/model_garden/requirements.txt
+++ b/generative_ai/model_garden/requirements.txt
@@ -2,7 +2,7 @@ pandas==2.2.3; python_version == '3.7'
 pandas==2.2.3; python_version == '3.8'
 pandas==2.2.3; python_version > '3.8'
 pillow==12.2.0
-google-cloud-aiplatform[full]==1.157.0
+google-cloud-aiplatform[full]==2.0.0
 sentencepiece==0.2.0
 google-auth==2.38.0
 anthropic[vertex]==0.28.0
diff --git a/generative_ai/model_tuning/requirements.txt b/generative_ai/model_tuning/requirements.txt
index 2bcc3b8bdc..4d46a9c26a 100644
--- a/generative_ai/model_tuning/requirements.txt
+++ b/generative_ai/model_tuning/requirements.txt
@@ -2,7 +2,7 @@ pandas==2.2.3; python_version == '3.7'
 pandas==2.2.3; python_version == '3.8'
 pandas==2.2.3; python_version > '3.8'
 pillow==12.2.0
-google-cloud-aiplatform[full]==1.157.0
+google-cloud-aiplatform[full]==2.0.0
 sentencepiece==0.2.0
 google-auth==2.38.0
 anthropic[vertex]==0.28.0
diff --git a/generative_ai/prompts/requirements.txt b/generative_ai/prompts/requirements.txt
index 2bcc3b8bdc..4d46a9c26a 100644
--- a/generative_ai/prompts/requirements.txt
+++ b/generative_ai/prompts/requirements.txt
@@ -2,7 +2,7 @@ pandas==2.2.3; python_version == '3.7'
 pandas==2.2.3; python_version == '3.8'
 pandas==2.2.3; python_version > '3.8'
 pillow==12.2.0
-google-cloud-aiplatform[full]==1.157.0
+google-cloud-aiplatform[full]==2.0.0
 sentencepiece==0.2.0
 google-auth==2.38.0
 anthropic[vertex]==0.28.0
diff --git a/generative_ai/provisioned_throughput/requirements.txt b/generative_ai/provisioned_throughput/requirements.txt
index a1bd1c36eb..64062981d8 100644
--- a/generative_ai/provisioned_throughput/requirements.txt
+++ b/generative_ai/provisioned_throughput/requirements.txt
@@ -1,2 +1,2 @@
-google-cloud-aiplatform==1.157.0
+google-cloud-aiplatform==2.0.0
 google-auth==2.38.0
diff --git a/generative_ai/rag/requirements.txt b/generative_ai/rag/requirements.txt
index 567a82bafe..61d1ddf6ff 100644
--- a/generative_ai/rag/requirements.txt
+++ b/generative_ai/rag/requirements.txt
@@ -1 +1 @@
-google-cloud-aiplatform==1.157.0
+google-cloud-aiplatform==2.0.0
diff --git a/generative_ai/reasoning_engine/requirements.txt b/generative_ai/reasoning_engine/requirements.txt
index 2bcc3b8bdc..4d46a9c26a 100644
--- a/generative_ai/reasoning_engine/requirements.txt
+++ b/generative_ai/reasoning_engine/requirements.txt
@@ -2,7 +2,7 @@ pandas==2.2.3; python_version == '3.7'
 pandas==2.2.3; python_version == '3.8'
 pandas==2.2.3; python_version > '3.8'
 pillow==12.2.0
-google-cloud-aiplatform[full]==1.157.0
+google-cloud-aiplatform[full]==2.0.0
 sentencepiece==0.2.0
 google-auth==2.38.0
 anthropic[vertex]==0.28.0
diff --git a/model_garden/gemma/requirements.txt b/model_garden/gemma/requirements.txt
index 1b2edafeb1..a81fb675a9 100644
--- a/model_garden/gemma/requirements.txt
+++ b/model_garden/gemma/requirements.txt
@@ -1 +1 @@
-google-cloud-aiplatform[full]==1.157.0
+google-cloud-aiplatform[full]==2.0.0
diff --git a/people-and-planet-ai/geospatial-classification/requirements.txt b/people-and-planet-ai/geospatial-classification/requirements.txt
index 2aae73b7a9..ffdaa03bd2 100644
--- a/people-and-planet-ai/geospatial-classification/requirements.txt
+++ b/people-and-planet-ai/geospatial-classification/requirements.txt
@@ -1,5 +1,5 @@
 earthengine-api==1.5.9
 folium==0.19.5
-google-cloud-aiplatform==1.157.0
+google-cloud-aiplatform==2.0.0
 pandas==2.2.3
 tensorflow==2.12.0
diff --git a/people-and-planet-ai/image-classification/requirements.txt b/people-and-planet-ai/image-classification/requirements.txt
index 4594256679..0f7a2c1ba6 100644
--- a/people-and-planet-ai/image-classification/requirements.txt
+++ b/people-and-planet-ai/image-classification/requirements.txt
@@ -1,3 +1,3 @@
 pillow==12.2.0
 apache-beam[gcp]==2.55.1
-google-cloud-aiplatform==1.157.0
+google-cloud-aiplatform==2.0.0
diff --git a/people-and-planet-ai/land-cover-classification/requirements.txt b/people-and-planet-ai/land-cover-classification/requirements.txt
index f4819f58b6..8f8eaf5faf 100644
--- a/people-and-planet-ai/land-cover-classification/requirements.txt
+++ b/people-and-planet-ai/land-cover-classification/requirements.txt
@@ -2,7 +2,7 @@
 apache-beam[gcp]==2.46.0
 earthengine-api==1.5.9
 folium==0.19.5
-google-cloud-aiplatform==1.157.0
+google-cloud-aiplatform==2.0.0
 imageio==2.36.1
 plotly==5.15.0
 tensorflow==2.12.0
diff --git a/people-and-planet-ai/timeseries-classification/requirements.txt b/people-and-planet-ai/timeseries-classification/requirements.txt
index 03aa507463..40dc4f674c 100644
--- a/people-and-planet-ai/timeseries-classification/requirements.txt
+++ b/people-and-planet-ai/timeseries-classification/requirements.txt
@@ -1,6 +1,6 @@
 Flask==3.1.3; python_version >= '3.9'
 apache-beam[gcp]==2.46.0
-google-cloud-aiplatform==1.157.0
+google-cloud-aiplatform==2.0.0
 gunicorn==23.0.0
 pandas==2.2.3
 tensorflow==2.12.1
diff --git a/people-and-planet-ai/weather-forecasting/tests/training_tests/requirements.txt b/people-and-planet-ai/weather-forecasting/tests/training_tests/requirements.txt
index d2b1b4ba08..6835328df5 100644
--- a/people-and-planet-ai/weather-forecasting/tests/training_tests/requirements.txt
+++ b/people-and-planet-ai/weather-forecasting/tests/training_tests/requirements.txt
@@ -1,3 +1,3 @@
 ../../serving/weather-model
 build==0.10.0
-google-cloud-aiplatform==1.157.0
+google-cloud-aiplatform==2.0.0
diff --git a/translate/samples/snippets/requirements.txt b/translate/samples/snippets/requirements.txt
index 9f532a6fc0..b5adb1010f 100644
--- a/translate/samples/snippets/requirements.txt
+++ b/translate/samples/snippets/requirements.txt
@@ -1,4 +1,4 @@
 google-cloud-translate==3.18.0
 google-cloud-storage==2.9.0
 google-cloud-automl==2.14.1
-google-cloud-aiplatform==1.157.0
+google-cloud-aiplatform==2.0.0