Arm 透過新的 PyTorch 和 ExecuTorch 整合加速從雲到邊緣的人工智慧協助開發人員立即提升效能

September 19, 2024

新聞重點:

  • Arm 透過將 Kleidi 技術整合到 PyTorch 和 ExecuTorch,使關鍵的 AI 效能優勢從邊緣端拓展至雲端,促使次世代的應用在 Arm CPU 上運行大型語言模型。
  • 對普及機器學習(ML)工作負載的持續投入將使任一技術堆疊的開發人員,能夠在最新的生成式 AI 模型上立即獲得顯著的推理效能提升。
  • 擴大與雲端服務供應商及主要的 ML 獨立軟體開發商的合作,進一步協助各地的 AI 開發人員。

Arm 控股有限公司(納斯達克股票代碼:ARM,以下簡稱 Arm)近期宣佈透過將 Arm® Kleidi 技術整合到 PyTorch 和 ExecuTorch,促使新一代的應用在 Arm CPU 上運行大型語言模型(LLM)。Kleidi 彙集了最新的開發人員賦能技術和關鍵資源,目標在於推動機器學習(ML)技術堆疊中的技術協作和創新。透過這些重要進展,Arm 致力於為每一位 ML 技術堆疊的開發人員提供更為順暢的體驗。

Arm 策略與生態部開發人員技術副總裁 Alex Spinelli 表示:「Arm 正與領先的雲端服務供應商和框架設計者緊密合作,以打造便捷的開發環境,讓軟體開發人員能夠輕鬆地在基於 Arm 架構的硬體上加速人工智慧(AI)和 ML 工作負載。自該技術推出的四個月以來,Kleidi 已在 Arm CPU 上加速開發並顯著提升主要的 AI 效能。Arm 與 PyTorch 社群的緊密合作,印證了該技術可大幅減少開發人員利用高效率 AI 所需的工作量。」

與領先框架整合,展現顯著的雲端優勢

在雲端,Kleidi 以利用 Arm Compute Library(ACL)強化 PyTorch 帶來的成果為基礎,為世界各地在 Arm 平台上優化 AI 的開發人員打造藍圖。透過為開發人員免去其不必要的工程工作,使開發人員能將 Arm 視為運行其關鍵 ML 工作負載的首選平台。做為實現此一願景的關鍵一步,Arm 直接與 PyTorch 和 TensorFlow 進行 Arm Kleidi Libraries(函式庫)的整合合作,這包括將基本的 Arm函式庫整合到上述的領先框架中。

重要的是,這代表著當新的框架版本發佈時,應用開發人員能夠自動地從其大幅的效能提升中獲益,而無需額外在 Arm 平台上進行建構。這項投入已對合作夥伴關係產生了正面的影響:

  • Arm 聊天機器人展示由 Meta Llama 3 LLM 所驅動,並運行在亞馬遜網路服務公司(AWS)Graviton 處理器上,首次在主線 PyTorch 中實現了即時聊天回應。
    • 根據在 AWS Graviton4 上所測得的資料顯示,透過將 Kleidi 技術整合到開源 PyTorch 程式庫,詞元(token)首次回應時間可加快 2.5 倍。
  • 透過優化 torch.compile 來更有效率地使用透過 ACL 提供的 Kleidi 技術,在基於 AWS Graviton3 上所測得的資料顯示,各類 Hugging Face 模型推理工作負載上的效能可提升 1.35 至 2 倍。

這些僅是出色的雲端範例之一,卻代表了在 Arm 平台上普及 ML 工作負載時可實現的效能加速類型。Arm 將持續投入,以確保開發人員的 AI 應用,能從雲到邊緣都完美運行,其中包括新功能與前代的相容,進而使得開發人員能夠立即從中獲益。

合作協助開發人員掌握生成式 AI 發展步伐

隨著新的語言模型版本快速地推陳出新,生成式 AI 掀起了一波 AI 創新熱潮。Arm 持續與 ML 技術堆疊的各個關鍵部分緊密合作,攜手 AWS 和 Google 等雲端服務供應商以及 Databricks 等迅速壯大的 ML 獨立軟體開發商(ISV)社群,進而協助開發人員立於技術的前端。

Google Cloud Compute 產品管理資深總監 Nirav Mehta 表示:「Arm 和 Google Cloud 致力於為開發人員提升 AI 的存取和敏捷性,而 Kleidi 代表了以軟硬體共同優化滿足 AI 需求所獲得的重要進展。隨著我們的客戶正積極採用基於 Arm 架構的客製化 CPU - Axion,我們期待在整個 ML 技術堆疊中為客戶帶來更為順暢的整合體驗。」

Databricks 軟體工程師 Lin Yuan 表示:「利用 Databricks Data Intelligence Platform 進行 AI 和 ML 工作流程的企業,將可受惠於跨 ML 軟體堆疊的 Arm Kleidi 整合帶來的效能優化。藉由 Databricks ML Runtime 叢集提供支援的 Arm 架構 AWS Graviton 處理器,企業可以從各種 ML 函式庫的加速中受惠,同時降低雲端服務供應商的成本。」

協助開發人員將 Arm 提供的資源應用到實際案例中至關重要,為此,Arm 創立範例軟體堆疊和學習資源,向開發人員展示如何在 Arm CPU 上建構 AI 工作負載,進而迅速推動了 Arm 系統的廣泛採用,並加快了開發人員在 Arm 系統上的部署速度。第一個案例是透過 Kleidi 技術加速聊天機器人的實現,今年稍晚,ML Ops 和檢索強化生成也將添加至這些應用案例中,並計畫於 2025 年實現更多成果。

持續提升裝置端效能

基於 Kleidi 在裝置端的發展態勢,KleidiAI 還將被整合到 ExecuTorch(PyTorch 新的裝置端推理運行runtime)。這項整合預計將於 2024 年 10 月完成,並有望為目前正在 ExecuTorch 中進行生產測試或推出的裝置端應用帶來顯著的效能提升。目前已完成的多項 KleidiAI 整合包括與 Google XNNPACK 和 MediaPipe,以及騰訊的混元大模型,為其實際工作負載帶來了顯著提升,其影響不言可喻,誠如此聊天機器人所展示

Kleidi 將繼續與 PyTorch 和 ExecuTorch 的各版本以及其他主要 AI 框架進行整合。從雲端資料中心到邊緣端裝置,開發人員現在可以即刻在各類裝置上基於 Arm 平台高效率的運行高效能 AI 工作負載。Arm 將繼續積極地為 PyTorch 社群推出強化功能,並專注於針對各種整數格式提供量化優化,進一步提高效能,使 Arm CPU 能大規模地無縫運行次世代的 AI 體驗。

實現更多成果以協助開發人員

PyTorch 正在大規模的推動 ML 開發領域的創新。近日,Arm 加入 PyTorch 基金會成為 Premier 成員,這對於 Arm 的 AI 之旅來說,無疑是一個重要時刻。Arm 將持續致力於協助全球各地的開發人員,在 Arm 平台上充分發揮端到端 AI 的潛力,進而塑造尖端的 AI 和應用功能。

附加資源:

Kleidi(古希臘語中意為「鑰匙」)基於三大關鍵支柱而建構:

  • 開放的 Arm 技術直接整合至主要框架中,開發人員無需任何額外工作,便能使 LLM 無縫取得 Arm CPU 效能。Arm 將確保新技術始終向前相容,以便開發人員可以立即從中獲益。
  • 透過提供使用指南、學習資源和技術展示等各種資源協助開發人員。
  • 藉由 ML 軟體供應商、框架和開源項目所組成的活力十足的生態系,從中取得各類最新的 AI 功能,讓 Arm 平台成為開發人員建構解決方案的首選平台。

關於 Arm

Arm 是業界效能最高且最節能的運算平台,其無可比擬的應用範疇觸及全球所有連網使用者。為因應全球對運算永無止境的需求,Arm 提供先進的解決方案,使全球領先的科技公司得以釋放前所未有的 AI 體驗與功能。透過與全球最大的運算生態系及 2,000 萬名軟體開發人員的共同努力,我們正在 Arm 平台上建構 AI 的未來。


所有資訊都「依目前情況」提供,且並不帶保證或代表性。此文件可以自由分享,但不得修改且必須註明出處。Arm 是 Arm Limited(或其子公司與附屬機構)的註冊商標。所有品牌或產品名稱均為所屬公司之財產。© 1995-2024 Arm Limited.