Update EnzymeJaX visibility

wsmoses · Google-ML-Automation · commit a80c856dc1a5 · 2025-12-10T18:48:55.000-08:00
PiperOrigin-RevId: 842981186
diff --git a/jaxlib/gpu/BUILD b/jaxlib/gpu/BUILD
@@ -82,6 +82,10 @@ proto_library(
 cc_proto_library(
     name = "triton_cc_proto",
     compatible_with = None,
+    visibility = [
+        "//jax:internal",
+        "//third_party/py/enzyme_ad:__subpackages__",
+    ],
     deps = [":triton_proto"],
 )
 
