
xla
openxla
A machine learning compiler for GPUs, CPUs, and ML accelerators
AI 简介
XLA(Accelerated Linear Algebra)是一个开源的机器学习编译器,用于将PyTorch、TensorFlow、JAX等框架的模型编译优化,提升其在GPU、CPU及专用AI加速器上的执行性能。它通过图级优化、算子融合、内存布局重排和硬件后端定制化代码生成等技术,显著降低延迟并提高吞吐量。适用于需要高性能推理或训练的生产环境,尤其适合对延迟敏感、资源受限或需跨平台部署的AI服务与边缘计算场景。
C++
4.5k
Stars
885
Forks
49
Watchers
237
Issues
Star 增长
今日0
近 7 天0
近 30 天0
综合评分30.84
默认分支main