From a722ec08f99bf88ce690a28604dc5af9c2911f53 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Fri, 10 Mar 2023 14:22:55 -0800
Subject: [PATCH] No changes.

PiperOrigin-RevId: 515737909
---
 jax/BUILD | 1 +
 1 file changed, 1 insertion(+)

diff --git a/jax/BUILD b/jax/BUILD
index dc8b13918..8a86a923f 100644
--- a/jax/BUILD
+++ b/jax/BUILD
@@ -315,6 +315,7 @@ pytype_library(
         "_src/distributed.py",
         "_src/xla_bridge.py",
     ],
+    visibility = [":internal"] + jax_visibility("xla_bridge"),
     deps = [
         ":cloud_tpu_init",
         ":config",