Internal

Google-ML-Automation · Google-ML-Automation · commit 04b062e4069d · 2025-01-30T05:00:15.000-08:00
PiperOrigin-RevId: 719441319
diff --git a/jax_tpu_embedding/sparsecore/lib/core/BUILD b/jax_tpu_embedding/sparsecore/lib/core/BUILD
@@ -17,7 +17,10 @@ load("//third_party/bazel/python:pytype.bzl", "pytype_strict_contrib_test", "pyt
 
 package(
     default_applicable_licenses = ["//:license"],
-    default_visibility = ["//jax_tpu_embedding/sparsecore:__subpackages__"],
+    default_visibility = [
+        "//jax_tpu_embedding/sparsecore:__subpackages__",
+        "//smartass/brain/configure/jax:__subpackages__",
+    ],
 )
 
 cc_library(
