Intel® Extension for OpenXLA* library
Project description
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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distributions
File details
Details for the file intel_extension_for_openxla-0.4.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.
File metadata
- Download URL: intel_extension_for_openxla-0.4.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
- Upload date:
- Size: 80.1 MB
- Tags: CPython 3.12, manylinux: glibc 2.17+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.10.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | b89fabf35a9be93db9ff4ff48ab53e49e34b064bb9c31bf67475a806a6fd9f58 |
|
| MD5 | 1a9530e7f1347182f772b1abc5968129 |
|
| BLAKE2b-256 | 93070dda1d60f4dab0eb68cd43f495b30e36ea56fb326acd6038badab228b402 |
File details
Details for the file intel_extension_for_openxla-0.4.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.
File metadata
- Download URL: intel_extension_for_openxla-0.4.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
- Upload date:
- Size: 80.1 MB
- Tags: CPython 3.11, manylinux: glibc 2.17+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.10.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 184b2ed0ffe5e07968afb4aad1198e4f9b21c36094e6aebdb4e2646044c95f08 |
|
| MD5 | 013ce2de81ba759f1ef87db22f4d070f |
|
| BLAKE2b-256 | fe31ce0cdfaca9769390c1bcd76b39f5ffbf30c41cf193b56072e903f086f4e1 |
File details
Details for the file intel_extension_for_openxla-0.4.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.
File metadata
- Download URL: intel_extension_for_openxla-0.4.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
- Upload date:
- Size: 80.1 MB
- Tags: CPython 3.10, manylinux: glibc 2.17+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.10.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 30cc5129b663b920a52dc30247a85f38e0e439598a4afec69f6e06497a14053e |
|
| MD5 | 3f1a473a0a58a0a621659145cb57cbe2 |
|
| BLAKE2b-256 | 990118ef487938bc5fc4c2ca45747ce889706ab1f7405d2567c6e3002ca7431d |
File details
Details for the file intel_extension_for_openxla-0.4.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.
File metadata
- Download URL: intel_extension_for_openxla-0.4.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
- Upload date:
- Size: 80.1 MB
- Tags: CPython 3.9, manylinux: glibc 2.17+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.10.14
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | bd0032a3d0588a7c13fd549ea40d152105d6a5c5efdfa268418f08d49ccaf5d8 |
|
| MD5 | e1e37a9492b077d194f2b0a93cb4649f |
|
| BLAKE2b-256 | e9e2003c7a35512308d20b4cd7e32eed105898e51e9afb1174df22cab0c02ddd |
