Awesome OpenXLA

OpenXLA 生態系統

OpenXLA 是開放生態系統，提供效能高、可攜帶且可擴充的機器學習 (ML) 基礎架構元件，可透過在前端架構和硬體後端之間分散工具，簡化 ML 開發作業。由 AI 建模、軟體和硬體領域的業界領導者打造。

社群如何使用 OpenXLA？本頁面整合了使用 OpenXLA 的存放區和專案連結，提供靈感和程式碼指標！

專案是否使用 OpenXLA？請傳送提取要求，並將其加入這個頁面！

架構

  • JAX 是一種機器學習架構，提供類似 NumPy 的 API，可用於編寫高效能機器學習模型
  • PyTorch/XLA 提供從 PyTorch 到 OpenXLA 和 StableHLO 的橋接
  • TensorFlow 是長期使用的機器學習架構，擁有龐大的生態系統

PJRT 外掛程式

  • libTPU 可讓模型在 Google Cloud TPU 上執行

邊緣編譯

  • Google AI Edge 會使用 StableHLO 做為輸入格式，並透過 LiteRT 部署至行動裝置。
    • AI Edge Torch 會透過 StableHLO 匯出 PyTorch 模型，以便在行動裝置上部署

工具和視覺化

  • Model Explorer 提供階層圖表視覺化功能，並支援 StableHLO 模型