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.7.0 | 0.5.0 | 0.5.0 |
| 0.6.0 | 0.4.38 | 0.4.38 |
| 0.5.0 | 0.4.30 | >= 0.4.30, <= 0.4.31 |
| 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
5 maintainers
Release files for intel-extension-for-openxla 0.7.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Built distributions (wheels)
| File | Reset | |||
|---|---|---|---|---|
| intel_extension_for_openxla-0.7.0-cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.whl | CPython 3.13 | CPython 3.13 | Linux glibc 2.17+ x86-64 | Details |
| intel_extension_for_openxla-0.7.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl | CPython 3.12 | CPython 3.12 | Linux glibc 2.17+ x86-64 | Details |
| intel_extension_for_openxla-0.7.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl | CPython 3.11 | CPython 3.11 | Linux glibc 2.17+ x86-64 | Details |
| intel_extension_for_openxla-0.7.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl | CPython 3.10 | CPython 3.10 | Linux glibc 2.17+ x86-64 | Details |
Total release size: 341.5 MB
Release files / intel_extension_for_openxla-0.7.0-cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
| Download URL | intel_extension_for_openxla-0.7.0-cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.whl |
|---|---|
| Size | 85.4 MB |
| Tags | CPython 3.13 Linux glibc 2.17+ x86-64 |
|
SHA-256 checksum How to use checksums |
02f6b82881ba50220a619f3c176d76a67c493c03be3983dc209817b6c249ed7c
|
|
BLAKE2b-256 checksum How to use checksums |
bc0012c640b6bacfc0da024f79c8c6685173037e4d7ef4d6ed359d9999b7c9b4
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.0.1 CPython/3.10.16
|
Release files / intel_extension_for_openxla-0.7.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
| Download URL | intel_extension_for_openxla-0.7.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl |
|---|---|
| Size | 85.4 MB |
| Tags | CPython 3.12 Linux glibc 2.17+ x86-64 |
|
SHA-256 checksum How to use checksums |
4f876dbf93e8d200d00fc9771b21e086398da8a0908ac6bb082fb0e88e2b8f3a
|
|
BLAKE2b-256 checksum How to use checksums |
c62ccf91c108d9480ef3f9866f7c0afbd567e25d9df75e90a5529ffa402f685a
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.0.1 CPython/3.10.16
|
Release files / intel_extension_for_openxla-0.7.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
| Download URL | intel_extension_for_openxla-0.7.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl |
|---|---|
| Size | 85.4 MB |
| Tags | CPython 3.11 Linux glibc 2.17+ x86-64 |
|
SHA-256 checksum How to use checksums |
abb2c9426cf9f9b016927375344d4162f1bdd1503dd436648efd2976fff76c76
|
|
BLAKE2b-256 checksum How to use checksums |
86ff4c5242a6622d0dc2ed3837591849667403653148434f021381ea75ab0c25
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.0.1 CPython/3.10.16
|
Release files / intel_extension_for_openxla-0.7.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
| Download URL | intel_extension_for_openxla-0.7.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl |
|---|---|
| Size | 85.4 MB |
| Tags | CPython 3.10 Linux glibc 2.17+ x86-64 |
|
SHA-256 checksum How to use checksums |
a844ffc2b4cc4f78f7610ce5cdd67c9f6c8772418a16bd8909a9e0b6c94c861d
|
|
BLAKE2b-256 checksum How to use checksums |
f64afd0c53605b0eaea94b795553381f820840cbc90c70a9211f89763ff295a4
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.0.1 CPython/3.10.16
|
