Avoid private import of jax._src.interpreters

PiperOrigin-RevId: 766689939

Author: Jake VanderPlas

pathwaysutils/plugin_executable.py

@@ -18,7 +18,7 @@
 from typing import List, Sequence, Tuple, Union
 
 import jax
-from jax._src.interpreters import pxla
+from jax.interpreters import pxla
 from jax.extend.ifrt_programs import ifrt_programs