Internal

PiperOrigin-RevId: 721350333
jax-ml · Jan 30, 2025 · 06fea4a · 06fea4a
1 parent b95d8bf
commit 06fea4a
Showing 1 changed file with 4 additions and 1 deletion.
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(