We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9fc2e96 commit 920985eCopy full SHA for 920985e
pathwaysutils/plugin_executable.py
@@ -19,7 +19,7 @@
19
20
import jax
21
from jax._src.interpreters import pxla
22
-from jax._src.lib.xla_extension import ifrt_programs
+from jaxlib.xla_extension import ifrt_programs
23
24
25
class PluginExecutable:
0 commit comments