IT资讯 TVM 成 ASF 顶级项目,多个大厂使用的深度学习硬件编译器堆栈

garrett · 2020-12-03 09:30:05 · 热度: 9

11 月 30 日,Apache 基金会宣布,Apache TVM 晋升顶级项目。TVM 是阿里云、AMD、ARM、AWS、Facebook、华为、英特尔、微软、NVIDIA 和 Xilinx 等厂商使用的开源端到端深度学习硬件编译器堆栈。

TVM 可用于 CPU、GPU 和专门的加速器,也是 ASF 首个全栈软件和硬件协同优化的项目。TVM 使机器学习开发人员能够在任何硬件后端有效地优化和运行计算。该项目始于 2017 年,是华盛顿大学的一个研究项目,2019 年 3 月进入 Apache 孵化器。

TVM 功能包括:

  • 部署:跨多种硬件模块,平台和系统(包括手机),可穿戴设备,专用芯片和嵌入式设备部署。
  • 高性能:编译和最小化的 runtimes 通常可解锁现有设备上的 ML 工作负载。
  • 灵活:支持 Keras,Apache MXNet(正在孵化),PyTorch,Tensorflow,CoreML 和 DarkNet 中的深度学习编译模型。
  • 多处运行:能够在后端,CPU,GPU,浏览器,微控制器,FPGA 和 ASIC 上运行。
  • 易用:支持 C ++,Rust,Java 和 Python。

猜你喜欢:
暂无回复。
需要 登录 后方可回复, 如果你还没有账号请点击这里 注册