From e4790b634eb6094ee365779f7b240d8c3a7d4303 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Fri, 20 Sep 2024 12:56:51 -0400
Subject: [PATCH] Don't pass --nocheck_visibility to Bazel.

This no longer appears to be needed.
---
 .bazelrc                | 4 ----
 jax_plugins/BUILD.bazel | 4 ++++
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/.bazelrc b/.bazelrc
index 5b7bc6533..f9f5cfc1b 100644
--- a/.bazelrc
+++ b/.bazelrc
@@ -1,10 +1,6 @@
 ############################################################################
 # All default build options below.
 
-# Required by OpenXLA
-# https://github.com/openxla/xla/issues/1323
-build --nocheck_visibility
-
 # Sets the default Apple platform to macOS.
 build --apple_platform_type=macos
 build --macos_minimum_os=10.14
diff --git a/jax_plugins/BUILD.bazel b/jax_plugins/BUILD.bazel
index 6e2cf6aad..08a590b09 100644
--- a/jax_plugins/BUILD.bazel
+++ b/jax_plugins/BUILD.bazel
@@ -12,6 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+package(
+    default_visibility = ["//jax:internal"],
+)
+
 licenses(["notice"])
 
 load(