diff --git a/stubs/tensorflow/tensorflow/__init__.pyi b/stubs/tensorflow/tensorflow/__init__.pyi
index 1021bff117a7..5817b9ad3811 100644
--- a/stubs/tensorflow/tensorflow/__init__.pyi
+++ b/stubs/tensorflow/tensorflow/__init__.pyi
@@ -11,6 +11,7 @@ from typing_extensions import ParamSpec, Self
 from google.protobuf.message import Message
 from tensorflow import (
     data as data,
+    distribute as distribute,
     experimental as experimental,
     feature_column as feature_column,
     initializers as initializers,
diff --git a/stubs/tensorflow/tensorflow/distribute/__init__.pyi b/stubs/tensorflow/tensorflow/distribute/__init__.pyi
new file mode 100644
index 000000000000..e6f7878c2ae7
--- /dev/null
+++ b/stubs/tensorflow/tensorflow/distribute/__init__.pyi
@@ -0,0 +1 @@
+from tensorflow.distribute import experimental as experimental