openxla/xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
解決的問題
XLA 解決了在不同硬體上高效執行機器學習模型的挑戰。它避免了為每個模型編寫特定硬體程式碼的需求，從而實現了在 GPU、CPU 和專用 ML 加速器上的高效能執行。
工作原理
XLA 作為一個編譯器，接收來自 PyTorch、TensorFlow 和 JAX 等流行框架的模型，並針對目標硬體平台進行專門優化，以確保高效能執行。
適用對象
此專案主要面向 ML 框架開發人員、希望開發該編譯器的 XLA 貢獻者，以及需要為新 ML 前端或硬體後端進行除錯或新增支援的整合人員。
亮點
- 支援包括 PyTorch、TensorFlow 和 JAX 在內的主流框架。
- 針對 CPU、GPU 和 ML 加速器的跨平台優化。
- 開源機器學習編譯器。
相關
- 專案
- 專案
- 專案
- 專案
- Dispatch