TACO обгоняет MATLAB Tensor Toolbox на порядки, генерируя ядра для разреженной тензорной алгебры

Taco: The Tensor Algebra Compiler

TACO — компилятор тензорной алгебры от исследовательских групп Stanford и MIT CSAIL. Он принимает разреженные и плотные вычисления — от умножения матрицы на вектор до MTTKRP на тензорах высших порядков — и генерирует оптимизированные ядра. Поддерживаются форматы CSR, COO и CSF, есть C++ и Python API. Производительность превышает MATLAB Tensor Toolbox на порядки и сравнима с Eigen, Intel oneMKL и SPLATT. Код открыт на GitHub, а генератор доступен прямо в браузере.