diff --git a/notebooks/official/sdk/sdk2_bigframes_pytorch.ipynb b/notebooks/official/vertex_ai_sdk/remote_training_bigframes_pytorch.ipynb
similarity index 98%
rename from notebooks/official/sdk/sdk2_bigframes_pytorch.ipynb
rename to notebooks/official/vertex_ai_sdk/remote_training_bigframes_pytorch.ipynb
index af208fa43..deef8fb3d 100644
--- a/notebooks/official/sdk/sdk2_bigframes_pytorch.ipynb
+++ b/notebooks/official/vertex_ai_sdk/remote_training_bigframes_pytorch.ipynb
@@ -33,18 +33,18 @@
         "\n",
         "<table align=\"left\">\n",
         "  <td>\n",
-        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/sdk/sdk2_bigframes_pytorch.ipynb\">\n",
+        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/vertex_ai_sdk/remote_training_bigframes_pytorch.ipynb\">\n",
         "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
         "    </a>\n",
         "  </td>\n",
         "  <td>\n",
-        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/sdk/sdk2_bigframes_pytorch.ipynb\">\n",
+        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/vertex_ai_sdk/remote_training_bigframes_pytorch.ipynb\">\n",
         "        <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
         "      View on GitHub\n",
         "    </a>\n",
         "  </td>\n",
         "    <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/sdk/sdk2_bigframes_pytorch.ipynb\">\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/vertex_ai_sdk/remote_training_bigframes_pytorch.ipynb\">\n",
         "       <img src=\"https://www.gstatic.com/cloud/images/navigation/vertex-ai.svg\" alt=\"Vertex AI logo\">Open in Vertex AI Workbench\n",
         "    </a>\n",
         "</table>"
@@ -517,6 +517,8 @@
       "outputs": [],
       "source": [
         "# define the custom model\n",
+        "\n",
+        "\n",
         "class TorchLogisticRegression(VertexModel, torch.nn.Module):\n",
         "    def __init__(self, input_size: int, output_size: int):\n",
         "        torch.nn.Module.__init__(self)\n",
@@ -692,7 +694,7 @@
   "metadata": {
     "colab": {
       "collapsed_sections": [],
-      "name": "sdk2_bigframes_pytorch.ipynb",
+      "name": "remote_training_bigframes_pytorch.ipynb",
       "toc_visible": true
     },
     "kernelspec": {
diff --git a/notebooks/official/sdk/sdk2_bigframes_sklearn.ipynb b/notebooks/official/vertex_ai_sdk/remote_training_bigframes_sklearn.ipynb
similarity index 98%
rename from notebooks/official/sdk/sdk2_bigframes_sklearn.ipynb
rename to notebooks/official/vertex_ai_sdk/remote_training_bigframes_sklearn.ipynb
index bbd4b31ae..11a93f331 100644
--- a/notebooks/official/sdk/sdk2_bigframes_sklearn.ipynb
+++ b/notebooks/official/vertex_ai_sdk/remote_training_bigframes_sklearn.ipynb
@@ -33,18 +33,18 @@
         "\n",
         "<table align=\"left\">\n",
         "  <td>\n",
-        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/sdk/sdk2_bigframes_sklearn.ipynb\">\n",
+        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/vertex_ai_sdk/remote_training_bigframes_sklearn.ipynb\">\n",
         "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
         "    </a>\n",
         "  </td>\n",
         "  <td>\n",
-        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/sdk/sdk2_bigframes_sklearn.ipynb\">\n",
+        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/vertex_ai_sdk/remote_training_bigframes_sklearn.ipynb\">\n",
         "        <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
         "      View on GitHub\n",
         "    </a>\n",
         "  </td>\n",
         "    <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/sdk/sdk2_bigframes_sklearn.ipynb\">\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/vertex_ai_sdk/remote_training_bigframes_sklearn.ipynb\">\n",
         "       <img src=\"https://www.gstatic.com/cloud/images/navigation/vertex-ai.svg\" alt=\"Vertex AI logo\">Open in Vertex AI Workbench\n",
         "    </a>\n",
         "</table>"
@@ -694,7 +694,7 @@
   "metadata": {
     "colab": {
       "collapsed_sections": [],
-      "name": "sdk2_bigframes_sklearn.ipynb",
+      "name": "remote_training_bigframes_sklearn.ipynb",
       "toc_visible": true
     },
     "kernelspec": {
diff --git a/notebooks/official/sdk/sdk2_bigframes_tensorflow.ipynb b/notebooks/official/vertex_ai_sdk/remote_training_bigframes_tensorflow.ipynb
similarity index 98%
rename from notebooks/official/sdk/sdk2_bigframes_tensorflow.ipynb
rename to notebooks/official/vertex_ai_sdk/remote_training_bigframes_tensorflow.ipynb
index 9bff4ba05..fc74bd846 100644
--- a/notebooks/official/sdk/sdk2_bigframes_tensorflow.ipynb
+++ b/notebooks/official/vertex_ai_sdk/remote_training_bigframes_tensorflow.ipynb
@@ -33,18 +33,18 @@
         "\n",
         "<table align=\"left\">\n",
         "  <td>\n",
-        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/sdk/sdk2_bigframes_tensorflow.ipynb\">\n",
+        "    <a href=\"https://colab.research.google.com/github/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/vertex_ai_sdk/remote_training_bigframes_tensorflow.ipynb\">\n",
         "      <img src=\"https://cloud.google.com/ml-engine/images/colab-logo-32px.png\" alt=\"Colab logo\"> Run in Colab\n",
         "    </a>\n",
         "  </td>\n",
         "  <td>\n",
-        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/sdk/sdk2_bigframes_tensorflow.ipynb\">\n",
+        "    <a href=\"https://github.com/GoogleCloudPlatform/vertex-ai-samples/blob/main/notebooks/official/vertex_ai_sdk/remote_training_bigframes_tensorflow.ipynb\">\n",
         "        <img src=\"https://cloud.google.com/ml-engine/images/github-logo-32px.png\" alt=\"GitHub logo\">\n",
         "      View on GitHub\n",
         "    </a>\n",
         "  </td>\n",
         "    <td>\n",
-        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/sdk/sdk2_bigframes_tensorflow.ipynb\">\n",
+        "    <a href=\"https://console.cloud.google.com/vertex-ai/workbench/deploy-notebook?download_url=https://raw.githubusercontent.com/GoogleCloudPlatform/vertex-ai-samples/main/notebooks/vertex_ai_sdk/remote_training_bigframes_tensorflow.ipynb\">\n",
         "       <img src=\"https://www.gstatic.com/cloud/images/navigation/vertex-ai.svg\" alt=\"Vertex AI logo\">Open in Vertex AI Workbench\n",
         "    </a>\n",
         "</table>"
@@ -627,7 +627,7 @@
   "metadata": {
     "colab": {
       "collapsed_sections": [],
-      "name": "sdk2_bigframes_tensorflow.ipynb",
+      "name": "remote_training_bigframes_tensorflow.ipynb",
       "toc_visible": true
     },
     "kernelspec": {