diff --git a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/createWorkspace.json b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/createWorkspace.json
index a01b4a0fc899..6ad8da73c875 100644
--- a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/createWorkspace.json
+++ b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/createWorkspace.json
@@ -11,7 +11,10 @@
         "tier": "Basic"
       },
       "identity": {
-        "type": "SystemAssigned"
+        "type": "SystemAssigned,UserAssigned",
+        "UserAssignedIdentities": {
+          "/subscriptions/00000000-1111-2222-3333-444444444444/resourceGroups/workspace-1234/providers/Microsoft.ManagedIdentity/userAssignedIdentities/testuai": {}
+        }
       },
       "properties": {
         "friendlyName": "HelloName",
@@ -22,6 +25,9 @@
         "storageAccount": "/subscriptions/00000000-1111-2222-3333-444444444444/resourceGroups/accountcrud-1234/providers/Microsoft.Storage/storageAccounts/testStorageAccount",
         "encryption": {
           "status": "Enabled",
+          "identity": {
+            "userAssignedIdentity": "/subscriptions/00000000-1111-2222-3333-444444444444/resourceGroups/workspace-1234/providers/Microsoft.ManagedIdentity/userAssignedIdentities/testuai"
+          },
           "keyVaultProperties": {
             "keyVaultArmId": "/subscriptions/00000000-1111-2222-3333-444444444444/resourceGroups/workspace-1234/providers/Microsoft.KeyVault/vaults/testkv",
             "keyIdentifier": "https://testkv.vault.azure.net/keys/testkey/aabbccddee112233445566778899aabb",
@@ -53,7 +59,14 @@
         "identity": {
           "principalId": "00000000-1111-2222-3333-444444444444",
           "tenantId": "00000000-1111-2222-3333-444444444444",
-          "type": "SystemAssigned"
+          "type": "SystemAssigned,UserAssigned",
+          "userAssignedIdentities": {
+            "/subscriptions/00000000-1111-2222-3333-444444444444/resourceGroups/workspace-1234/providers/Microsoft.ManagedIdentity/userAssignedIdentities/testuai": {
+              "principalId": "00000000-1111-2222-3333-444444444444",
+              "clientId": "00000000-1111-2222-3333-444444444444",
+              "tenantId": "00000000-1111-2222-3333-444444444444"
+            }
+          }
         },
         "sku": {
           "name": "Basic",
@@ -70,6 +83,9 @@
           "description": "test description",
           "encryption": {
             "status": "Enabled",
+            "identity": {
+              "userAssignedIdentity": "/subscriptions/00000000-1111-2222-3333-444444444444/resourceGroups/workspace-1234/providers/Microsoft.ManagedIdentity/userAssignedIdentities/testuai"
+            },
             "keyVaultProperties": {
               "keyVaultArmId": "/subscriptions/00000000-1111-2222-3333-444444444444/resourceGroups/workspace-1234/providers/Microsoft.KeyVault/vaults/testkv",
               "keyIdentifier": "https://testkv.vault.azure.net/keys/testkey/aabbccddee112233445566778899aabb",
@@ -98,6 +114,18 @@
         "name": "testworkspace",
         "type": "Microsoft.MachineLearningServices/workspaces",
         "location": "eastus2euap",
+        "identity": {
+          "principalId": "00000000-1111-2222-3333-444444444444",
+          "tenantId": "00000000-1111-2222-3333-444444444444",
+          "type": "SystemAssigned,UserAssigned",
+          "userAssignedIdentities": {
+            "/subscriptions/00000000-1111-2222-3333-444444444444/resourceGroups/workspace-1234/providers/Microsoft.ManagedIdentity/userAssignedIdentities/testuai": {
+              "principalId": "00000000-1111-2222-3333-444444444444",
+              "clientId": "00000000-1111-2222-3333-444444444444",
+              "tenantId": "00000000-1111-2222-3333-444444444444"
+            }
+          }
+        },
         "sku": {
           "name": "Basic",
           "tier": "Basic"
@@ -113,6 +141,9 @@
           "description": "test description",
           "encryption": {
             "status": "Enabled",
+            "identity": {
+              "userAssignedIdentity": "/subscriptions/00000000-1111-2222-3333-444444444444/resourceGroups/workspace-1234/providers/Microsoft.ManagedIdentity/userAssignedIdentities/testuai"
+            },
             "keyVaultProperties": {
               "keyVaultArmId": "/subscriptions/00000000-1111-2222-3333-444444444444/resourceGroups/workspace-1234/providers/Microsoft.KeyVault/vaults/testkv",
               "keyIdentifier": "https://testkv.vault.azure.net/keys/testkey/aabbccddee112233445566778899aabb",
diff --git a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/getWorkspace.json b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/getWorkspace.json
index b135a390d261..b60db18a58b2 100644
--- a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/getWorkspace.json
+++ b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/getWorkspace.json
@@ -12,6 +12,18 @@
         "name": "testworkspace",
         "type": "Microsoft.MachineLearningServices/workspaces",
         "location": "eastus2euap",
+        "identity": {
+          "principalId": "00000000-1111-2222-3333-444444444444",
+          "tenantId": "00000000-1111-2222-3333-444444444444",
+          "type": "SystemAssigned,UserAssigned",
+          "userAssignedIdentities": {
+            "/subscriptions/00000000-1111-2222-3333-444444444444/resourceGroups/workspace-1234/providers/Microsoft.ManagedIdentity/userAssignedIdentities/testuai": {
+              "principalId": "00000000-1111-2222-3333-444444444444",
+              "clientId": "00000000-1111-2222-3333-444444444444",
+              "tenantId": "00000000-1111-2222-3333-444444444444"
+            }
+          }
+        },
         "sku": {
           "name": "Enterprise",
           "tier": "Enterprise"
@@ -27,6 +39,9 @@
           "description": "test description",
           "encryption": {
             "status": "Enabled",
+            "identity": {
+              "userAssignedIdentity": "/subscriptions/00000000-1111-2222-3333-444444444444/resourceGroups/workspace-1234/providers/Microsoft.ManagedIdentity/userAssignedIdentities/testuai"
+            },
             "keyVaultProperties": {
               "keyVaultArmId": "/subscriptions/00000000-1111-2222-3333-444444444444/resourceGroups/workspace-1234/providers/Microsoft.KeyVault/vaults/testkv",
               "keyIdentifier": "https://testkv.vault.azure.net/keys/testkey/aabbccddee112233445566778899aabb",
diff --git a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/listNotebookAccessToken.json b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/listNotebookAccessToken.json
index f08e1971557b..991a9a50c33b 100644
--- a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/listNotebookAccessToken.json
+++ b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/listNotebookAccessToken.json
@@ -1,22 +1,22 @@
 {
-    "parameters": {
-      "subscriptionId": "00000000-1111-2222-3333-444444444444",
-      "resourceGroupName": "workspace-1234",
-      "workspaceName": "testworkspace",
-      "api-version": "2020-09-01-preview"
-    },
-    "responses": {
-      "200": {
-        "body": {
-          "notebookResourceId": "94350843095843059",
-          "hostName": "Host product name",
-          "publicDns": "resource.notebooks.azure.net",
-          "accessToken": "token",
-          "tokenType": "Bearer",
-          "expiresIn": 28800.0,
-          "refreshToken": "refresh token",
-          "scope": "aznb_identity"
-        }
+  "parameters": {
+    "subscriptionId": "00000000-1111-2222-3333-444444444444",
+    "resourceGroupName": "workspace-1234",
+    "workspaceName": "testworkspace",
+    "api-version": "2020-09-01-preview"
+  },
+  "responses": {
+    "200": {
+      "body": {
+        "notebookResourceId": "94350843095843059",
+        "hostName": "Host product name",
+        "publicDns": "resource.notebooks.azure.net",
+        "accessToken": "token",
+        "tokenType": "Bearer",
+        "expiresIn": 28800.0,
+        "refreshToken": "refresh token",
+        "scope": "aznb_identity"
       }
     }
-  }
\ No newline at end of file
+  }
+}
diff --git a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/listStorageAccountKeys.json b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/listStorageAccountKeys.json
index 122f8ed7341f..53487456a2dd 100644
--- a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/listStorageAccountKeys.json
+++ b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/listStorageAccountKeys.json
@@ -1,16 +1,15 @@
 {
-    "parameters": {
-      "subscriptionId": "aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee",
-      "resourceGroupName": "testrg123",
-      "workspaceName": "workspaces123",
-      "api-version": "2021-01-01"
-    },
-    "responses": {
-      "200": {
-        "body": {
-          "userStorageKey": "<userStorageKey>"
-        }
+  "parameters": {
+    "subscriptionId": "aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee",
+    "resourceGroupName": "testrg123",
+    "workspaceName": "workspaces123",
+    "api-version": "2021-01-01"
+  },
+  "responses": {
+    "200": {
+      "body": {
+        "userStorageKey": "<userStorageKey>"
       }
     }
   }
-  
\ No newline at end of file
+}
diff --git a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/updateWorkspace.json b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/updateWorkspace.json
index 4814280b0af9..9c65bb2df993 100644
--- a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/updateWorkspace.json
+++ b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/examples/updateWorkspace.json
@@ -22,6 +22,11 @@
         "name": "testworkspace",
         "type": "Microsoft.MachineLearningServices/workspaces",
         "location": "eastus2euap",
+        "identity": {
+          "principalId": "00000000-1111-2222-3333-444444444444",
+          "tenantId": "00000000-1111-2222-3333-444444444444",
+          "type": "SystemAssigned"
+        },
         "sku": {
           "name": "Enterprise",
           "tier": "Enterprise"
diff --git a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/machineLearningServices.json b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/machineLearningServices.json
index a2ca4357bddb..d1ff0d23d300 100644
--- a/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/machineLearningServices.json
+++ b/specification/machinelearningservices/resource-manager/Microsoft.MachineLearningServices/stable/2021-01-01/machineLearningServices.json
@@ -4655,6 +4655,10 @@
             "modelAsString": true
           }
         },
+        "identity": {
+          "$ref": "#/definitions/IdentityForCmk",
+          "description": "The identity that will be used to access the key vault for encryption at rest."
+        },
         "keyVaultProperties": {
           "$ref": "#/definitions/KeyVaultProperties",
           "description": "Customer Key vault properties."
@@ -4687,6 +4691,19 @@
       ],
       "type": "object"
     },
+    "IdentityForCmk": {
+      "description": "Identity that will be used to access key vault for encryption at rest",
+      "type": "object",
+      "properties": {
+        "userAssignedIdentity": {
+          "description": "The ArmId of the user assigned identity that will be used to access the customer managed key vault",
+          "type": "string"
+        }
+      },
+      "required": [
+        "userAssignedIdentity"
+      ]
+    },
     "LinkedServiceResponse": {
       "description": "Linked service.",
       "type": "object",