谷歌宣佈開源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框架、硬件平臺、用戶和集成商的代表。