jax icon indicating copy to clipboard operation
jax copied to clipboard
verified publisher icon google

[XLA:Python] Add support for executing portable PJRT executables from Python.

Open copybara-service[bot] opened this issue 3 years ago • 0 comments

[XLA:Python] Add support for executing portable PJRT executables from Python.

Add an optional device argument to execute that allows the user to trigger the ExecutePortable path. Plumb the compile_portable_executable compilation option through to Python.

[JAX] Handle compile_portable_executable option in compilation cache.

copybara-service[bot] avatar Aug 11 '22 15:08 copybara-service[bot]