XLA

XLA (加速線性代數) 是機器學習的開放原始碼編譯器。XLA 編譯器會從 PyTorch、TensorFlow 和 JAX 等熱門架構中取得模型，並針對 GPU、CPU 和機器學習加速器等不同硬體平台，為模型進行最佳化，以便在這些平台上執行高效能運算。

XLA 是 OpenXLA 專案的一部分，由業界領先的 ML 硬體和軟體公司共同建構，包括 Alibaba、Amazon Web Services、AMD、Apple、Arm、Google、Intel、Meta 和 NVIDIA。

主要優點

• 隨處建構：XLA 已整合至 TensorFlow、PyTorch 和 JAX 等頂尖機器學習架構。

• 可在任何地方執行：支援各種後端，包括 GPU、CPU 和 ML 加速器，並提供可插入式基礎架構，可新增更多支援。

• 提升效能並擴大規模：透過經過實際測試的最佳化階段，以及自動劃分模型並行處理，提升模型效能。

• 消除複雜性：此工具可運用 MLIR 的強大功能，將最佳功能整合至單一編譯器工具鍊，因此您不必管理一系列特定領域的編譯器。

• 未來就緒：XLA 是透過領先的機器學習硬體和軟體供應商合作建構的開放原始碼專案，旨在以最先進的機器學習技術運作。

說明文件

如要進一步瞭解 XLA，請參閱左側的連結。如果您是 XLA 新手開發人員，建議您先參閱「XLA 架構」，然後再閱讀「貢獻」一文。