Skip to main content

Intel® Extension for OpenXLA* library

Project description

Intel® Extension for OpenXLA*

Python PyPI version version

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

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

intel_extension_for_openxla-0.4.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (80.1 MB view hashes)

Uploaded CPython 3.12 manylinux: glibc 2.17+ x86-64

intel_extension_for_openxla-0.4.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (80.1 MB view hashes)

Uploaded CPython 3.11 manylinux: glibc 2.17+ x86-64

intel_extension_for_openxla-0.4.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (80.1 MB view hashes)

Uploaded CPython 3.10 manylinux: glibc 2.17+ x86-64

intel_extension_for_openxla-0.4.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (80.1 MB view hashes)

Uploaded CPython 3.9 manylinux: glibc 2.17+ x86-64

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page