[bazel] add visibility hook for //jax/_src:partial_eval

PiperOrigin-RevId: 800643271

Author: Jake VanderPlas

jax/_src/BUILD

@@ -1012,6 +1012,7 @@ pytype_strict_library(
 pytype_strict_library(
     name = "partial_eval",
     srcs = ["interpreters/partial_eval.py"],
+    visibility = ["//jax:internal"] + jax_visibility("partial_eval"),
     deps = [
         ":ad_util",
         ":api_util",