openxla

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.4k
Stars
857
Forks
51
Watchers
230
Issues

Star 增长

今日0
近 7 天0
近 30 天+14
综合评分59.2
默认分支main