
tvm
apache
Open Machine Learning Compiler Framework
AI 简介
Apache TVM 是一个开源的机器学习编译器框架,用于将深度学习模型自动优化并部署到多样化的硬件后端。其核心功能包括:基于Python的可编程编译流水线、多级IR(如Relay、TIR)抽象、支持GPU(CUDA、Vulkan、Metal、OpenCL、ROCm)、CPU及嵌入式设备的统一代码生成与性能调优。技术特点涵盖算子自动调度、内存布局优化、量化感知编译及跨平台部署能力。适用于AI模型在边缘设备、移动端、服务器或异构加速器上的高效推理与落地场景。
Python
Apache License 2.013.6k
Stars
3.9k
Forks
365
Watchers
131
Issues
Star 增长
今日0
近 7 天0
近 30 天+51
综合评分75.1
默认分支main