intel-extension-for-openxla 0.4.0

Intel® Extension for OpenXLA* library

Intel® Extension for OpenXLA*

The OpenXLA Project brings together a community of developers and leading AI/ML teams to accelerate ML and address infrastructure fragmentation across ML frameworks and hardware.

Intel® Extension for OpenXLA* includes PJRT plugin implementation, which seamlessly runs JAX models on Intel GPU. The PJRT API simplified the integration, which allowed the Intel GPU plugin to be developed separately and quickly integrated into JAX.

Installation

The following table tracks intel-extension-for-openxla versions and compatible versions of jax, jaxlib.

intel-extension-for-openxla	jaxlib	jax
0.4.0	0.4.26	>= 0.4.26, <= 0.4.27
0.3.0	0.4.24	>= 0.4.24, <= 0.4.27
0.2.1	0.4.20	>= 0.4.20, <= 0.4.26
0.2.0	0.4.20	>= 0.4.20, <= 0.4.26
0.1.0	0.4.13	>= 0.4.13, <= 0.4.14

pip install --upgrade intel-extension-for-openxla

Security

See Intel's Security Center for information on how to report a potential security issue or vulnerability.

See also: Security Policy