跳到正文
Hacker News· measurablefunc·· 2 小时前AI 评分25

TACO:张量代数编译器

Taco: The Tensor Algebra Compiler

AI 导读

TACO 是斯坦福与 MIT CSAIL 研究人员开发的张量代数编译器,可将稀疏线性与张量代数计算自动生成优化内核。它支持 CSR、COO、CSF 等存储格式,性能最高比 MATLAB Tensor Toolbox 快数个数量级,并与 Eigen、Intel oneMKL、SPLATT 等库相当。TACO 提供 C++ 和 Python API 及命令行与网页工具,代码托管于 GitHub。

来源:Hacker News · tensor-compiler.org