diff --git a/jax_tpu_embedding/sparsecore/lib/flax/BUILD b/jax_tpu_embedding/sparsecore/lib/flax/BUILD
index 43c49ff8..3385ef7f 100644
--- a/jax_tpu_embedding/sparsecore/lib/flax/BUILD
+++ b/jax_tpu_embedding/sparsecore/lib/flax/BUILD
@@ -18,7 +18,6 @@ package(
     default_applicable_licenses = ["//:license"],
     default_visibility = [
         "//jax_tpu_embedding/sparsecore:__subpackages__",
-        pypi_requirement("recml:__subpackages__"),
         "//video/youtube/discovery/lrm/plum:__subpackages__",
     ],
 )
diff --git a/jax_tpu_embedding/sparsecore/lib/nn/BUILD b/jax_tpu_embedding/sparsecore/lib/nn/BUILD
index eb0849ab..6355a0b4 100644
--- a/jax_tpu_embedding/sparsecore/lib/nn/BUILD
+++ b/jax_tpu_embedding/sparsecore/lib/nn/BUILD
@@ -18,7 +18,6 @@ package(
     default_applicable_licenses = ["//:license"],
     default_visibility = [
         "//jax_tpu_embedding/sparsecore:__subpackages__",
-        pypi_requirement("recml:__subpackages__"),
         "//video/youtube/discovery/lrm/plum:__subpackages__",
         "//video/youtube/discovery/ml_infra/jax:__subpackages__",
     ],
diff --git a/jax_tpu_embedding/sparsecore/utils/BUILD b/jax_tpu_embedding/sparsecore/utils/BUILD
index f72773a2..9d9a6a89 100644
--- a/jax_tpu_embedding/sparsecore/utils/BUILD
+++ b/jax_tpu_embedding/sparsecore/utils/BUILD
@@ -16,10 +16,7 @@ load("//third_party/bazel/python:pytype.bzl", "pytype_strict_library")
 
 package(
     default_applicable_licenses = ["//:license"],
-    default_visibility = [
-        "//jax_tpu_embedding/sparsecore:__subpackages__",
-        pypi_requirement("recml:__subpackages__"),
-    ],
+    default_visibility = ["//jax_tpu_embedding/sparsecore:__subpackages__"],
 )
 
 pytype_strict_library(