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