谷歌宣布开源OpenXLA

在Next ' 22活动上，谷歌Cloud宣布推出开源机器学习（ML）编译器生态系统OpenXLA。

OpenXLA是一个由社区主导的开源生态系统，由谷歌和其他AI（人工智能）/ML（机器学习）开发者(包括AMD、Arm、Meta、NVIDIA、AWS、Intel和Apple)共同开发的ML编译器和基础设施项目。

框架和硬件之间的不兼容可能会阻碍ML开发，而OpenXLA的目标是通过为开发人员在为ML项目选择框架和硬件时提供灵活性来解决这个问题。

根据OpenXLA项目章程，该项目的目标是通过与主要ML框架和硬件供应商的协作，实现从大多数主流框架(如PyTorch和TensorFlow)到任何硬件后端(特别是cpu、gpu和ML asic)的ML模型的高效降低、优化和部署。

谷歌副总裁兼基础设施总经理Sachin Gupta在一篇博客文章中解释说，新社区项目的第一个目标将是XLA编译器的协同进化，XLA编译器的开发是为了通过加速训练过程和减少总体内存消耗来简化TensorFlow中的建模。编译器现在正从TensorFlow中解耦，OpenXLA将致力于构建StableHLO，这是一个可移植的ML计算操作集，充当机器学习框架和编译器之间的可移植层。

OpenXLA的目标如下:

加速围绕XLA的行业协作，并构建一个充满活力的OSS社区。

分享和接收关于OpenXLA技术方向的反馈，并确保它满足主要用户和贡献者的需求。

建立一个新的XLA存储库或组织，使用独立的构建/测试，使用更容易接受PRs的基础设施，这是独立于硬件和框架的。

确保从TensorFlow提取XLA对现有用户和贡献者的干扰最小。

用自己的品牌、网站、文档和沟通渠道创建产品标识。

讨论在TensorFlow之外建立治理。

“在谷歌，我们相信开源软件对于克服与不灵活战略相关的挑战是必不可少的。作为云原生计算基金会的主要贡献者，我们有超过20年的经验与社区合作，将OSS项目变成技术进步的可访问的、透明的催化剂。我们致力于所有类型的开放生态系统，这一承诺延伸到AI/ML——我们坚信没有一家公司应该拥有AI/ML创新，”Gupta说。

OpenXLA的会员资格向所有参与开发或集成XLA的人开放，包括ML框架、硬件平台、用户和集成商的代表。