This CL introduces 'PluginProgram' in IFRT and exposes this in python via xla_client.compile_ifrt_program().

The IFRT `PluginProgram` is simply a wrapper for arbitrary byte-strings: an IFRT backend that recognizes `PluginProgram` can interpret the byte-string in any way it sees fit.

PiperOrigin-RevId: 621258245

Author: jax authors

jaxlib/tools/build_wheel.py

@@ -100,6 +100,7 @@ def patch_copy_mlir_import(src_file, dst_dir):
 
 _XLA_EXTENSION_STUBS = [
     "__init__.pyi",
+    "ifrt_programs.pyi",
     "ifrt_proxy.pyi",
     "jax_jit.pyi",
     "ops.pyi",