JAX-QNN 发布: JAX 可直连高通 Hexagon NPU

carrycooldude · x · 2026-08-15

开发者发布了 JAX-QNN,这是一个开源的原生 JAX 后端,旨在让 JAX 计算直接运行在高通 Snapdragon 硬件加速器(包括 Hexagon NPU、Adreno GPU 和 Oryon CPU)上。通过 OpenXLA PJRT C API 实现,标准 JAX 代码可编译为原生高通执行图。在 Snapdragon X Elite (45 TOPS NPU) 上的基准测试显示,Transformer 自注意力等任务相比 CPU 有 5 倍以上的加速。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →