跳转到主要内容

Intel® OpenXLA*库扩展

项目描述

Intel® OpenXLA*库扩展

Python PyPI version version

OpenXLA项目汇集了一群开发者和领先的AI/ML团队,以加速机器学习并解决ML框架和硬件之间的基础设施碎片化问题。

Intel® 对 OpenXLA* 的扩展包括 PJRT 插件实现,它能够在 Intel GPU 上无缝运行 JAX 模型。PJRT API 简化了集成过程,使得 Intel GPU 插件可以独立开发并快速集成到 JAX。

安装

以下表格跟踪 intel-extension-for-openxla 版本和 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

安全

有关如何报告潜在的安全问题或漏洞的信息,请参阅 Intel 的 安全中心

另请参阅:安全策略

支持者