diff --git a/db/docs/ml_candidates.yml b/db/docs/ml_candidates.yml
index a79729e154841954753b34ed4c0a22c8508c7ef8..6c17cf22dd30c080f811ad82a042d33147b5f855 100644
--- a/db/docs/ml_candidates.yml
+++ b/db/docs/ml_candidates.yml
@@ -1,10 +1,26 @@
 ---
 table_name: ml_candidates
 classes:
-  - Ml::Candidate
+- Ml::Candidate
 feature_categories:
-  - mlops
-description: A Model Candidate is a record of the results on training a model on some configuration
+- mlops
+description: A Model Candidate is a record of the results on training a model on some
+  configuration
 introduced_by_url: https://gitlab.com/gitlab-org/gitlab/-/merge_requests/95168
 milestone: '15.4'
-gitlab_schema: gitlab_main
+gitlab_schema: gitlab_main_cell
+allow_cross_joins:
+- gitlab_main_clusterwide
+allow_cross_transactions:
+- gitlab_main_clusterwide
+allow_cross_foreign_keys:
+- gitlab_main_clusterwide
+desired_sharding_key:
+  project_id:
+    references: projects
+    backfill_via:
+      parent:
+        foreign_key: experiment_id
+        table: ml_experiments
+        sharding_key: project_id
+        belongs_to: experiment
diff --git a/db/docs/ml_experiment_metadata.yml b/db/docs/ml_experiment_metadata.yml
index 812a33fe176cfe4ace3aedcb62abcd27b7aba8db..7b144b3dd986214d0b87e396438569a91d25fab6 100644
--- a/db/docs/ml_experiment_metadata.yml
+++ b/db/docs/ml_experiment_metadata.yml
@@ -1,10 +1,25 @@
 ---
 table_name: ml_experiment_metadata
 classes:
-  - Ml::ExperimentMetadata
+- Ml::ExperimentMetadata
 feature_categories:
-  - mlops
-gitlab_schema: gitlab_main
+- mlops
+gitlab_schema: gitlab_main_cell
 description: An Experiment Metadata record holds extra information about the experiment
 introduced_by_url: https://gitlab.com/gitlab-org/gitlab/-/merge_requests/104267
 milestone: '15.7'
+allow_cross_joins:
+- gitlab_main_clusterwide
+allow_cross_transactions:
+- gitlab_main_clusterwide
+allow_cross_foreign_keys:
+- gitlab_main_clusterwide
+desired_sharding_key:
+  project_id:
+    references: projects
+    backfill_via:
+      parent:
+        foreign_key: experiment_id
+        table: ml_experiments
+        sharding_key: project_id
+        belongs_to: experiment
diff --git a/db/docs/ml_model_metadata.yml b/db/docs/ml_model_metadata.yml
index 0f48f71dcbb37b8da9ca457c08fff8c81903b867..141d17d73bbffd229a30af7562ac300d9260d618 100644
--- a/db/docs/ml_model_metadata.yml
+++ b/db/docs/ml_model_metadata.yml
@@ -1,10 +1,25 @@
 ---
 table_name: ml_model_metadata
 classes:
-  - Ml::ModelMetadata
+- Ml::ModelMetadata
 feature_categories:
-  - mlops
-gitlab_schema: gitlab_main
+- mlops
+gitlab_schema: gitlab_main_cell
 description: A Model Metadata record holds extra information about the model
 introduced_by_url: https://gitlab.com/gitlab-org/gitlab/-/merge_requests/134443
 milestone: 16.6
+allow_cross_joins:
+- gitlab_main_clusterwide
+allow_cross_transactions:
+- gitlab_main_clusterwide
+allow_cross_foreign_keys:
+- gitlab_main_clusterwide
+desired_sharding_key:
+  project_id:
+    references: projects
+    backfill_via:
+      parent:
+        foreign_key: model_id
+        table: ml_models
+        sharding_key: project_id
+        belongs_to: model