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