diff --git a/WORKSPACE b/WORKSPACE
index 3f83edd20..4c9df3df0 100644
--- a/WORKSPACE
+++ b/WORKSPACE
@@ -184,6 +184,8 @@ http_archive(
     patches = [
         "//third_party:tensorflow.patch",
         "//third_party:tensorflow_add_grpc_cares_darwin_arm64_support.patch",
+        "//third_party:tensorflow_public_visibility.patch",  # TODO(matthurd): Remove after tf migration.
+        "//third_party:tensorflow_public_visibility_2.patch",  # TODO(matthurd): Remove after tf migration.
     ],
     strip_prefix = "tensorflow-master",
     urls = [
diff --git a/third_party/tensorflow_public_visibility.patch b/third_party/tensorflow_public_visibility.patch
new file mode 100644
index 000000000..028a1e5d5
--- /dev/null
+++ b/third_party/tensorflow_public_visibility.patch
@@ -0,0 +1,11 @@
+--- a/tensorflow/core/profiler/convert/BUILD
++++ b/tensorflow/core/profiler/convert/BUILD
+@@ -4,7 +4,7 @@
+
+ package(
+     # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],
+-    default_visibility = ["//tensorflow/core/profiler:internal"],
++    default_visibility = ["//visibility:public"],
+     licenses = ["notice"],
+ )
+
diff --git a/third_party/tensorflow_public_visibility_2.patch b/third_party/tensorflow_public_visibility_2.patch
new file mode 100644
index 000000000..9ce204c93
--- /dev/null
+++ b/third_party/tensorflow_public_visibility_2.patch
@@ -0,0 +1,11 @@
+--- a/tensorflow/core/profiler/protobuf/BUILD
++++ b/tensorflow/core/profiler/protobuf/BUILD
+@@ -33,7 +33,7 @@
+     protodeps = [
+         "@local_tsl//tsl/profiler/protobuf:xplane_proto",
+     ],
+-    visibility = [":friends"],
++    visibility = ["//visibility:public"],
+     exports = [
+         "@local_tsl//tsl/profiler/protobuf:xplane_proto",
+     ],
diff --git a/xprof/BUILD b/xprof/BUILD
new file mode 100644
index 000000000..e69de29bb
diff --git a/xprof/__init__.py b/xprof/__init__.py
new file mode 100644
index 000000000..e69de29bb