Add experimental/serialize_executable.py to BUILD.

PiperOrigin-RevId: 736975882

jax/BUILD

@@ -36,6 +36,7 @@ load(
     "py_library_providing_imports_info",
     "pytype_library",
     "pytype_strict_library",
+    "serialize_executable_internal_users",
 )
 
 package(
@@ -127,6 +128,12 @@ package_group(
     packages = mosaic_gpu_internal_users,
 )
 
+package_group(
+    name = "serialize_executable_users",
+    includes = [":internal"],
+    packages = serialize_executable_internal_users,
+)
+
 # JAX-private test utilities.
 py_library(
     # This build target is required in order to use private test utilities in jax._src.test_util,
@@ -618,6 +625,16 @@ pytype_strict_library(
     ] + py_deps("numpy"),
 )
 
+pytype_strict_library(
+    name = "serialize_executable",
+    srcs = ["experimental/serialize_executable.py"],
+    visibility = [":serialize_executable_users"],
+    deps = [
+        ":jax",
+        "//jax/_src/lib",
+    ],
+)
+
 pytype_strict_library(
     name = "source_mapper",
     srcs = glob(include = ["experimental/source_mapper/**/*.py"]),

jaxlib/jax.bzl

@@ -48,6 +48,7 @@ pallas_gpu_internal_users = []
 pallas_tpu_internal_users = []
 pallas_fuser_users = []
 mosaic_extension_deps = []
+serialize_executable_internal_users = []
 
 jax_internal_export_back_compat_test_util_visibility = []
 jax_internal_test_harnesses_visibility = []