Sep, 2023

CoLA:利用组合结构进行自动和高效的数值线性代数

TL;DR该研究提出了一种名为 CoLA(组合线性代数)的大规模线性代数问题的简单而通用的框架,通过将线性操作符抽象与组合调度规则相结合,CoLA 自动构建了内存和运行时高效的数值算法,同时提供内存高效的自动微分、低精度计算和 JAX 和 PyTorch 的 GPU 加速,还通过多重调度在下游包中允许新对象、操作和规则。CoLA 可以加速许多代数操作,同时使得原型化矩阵结构和算法变得简单,可适用于几乎任何需要线性代数的计算任务。我们展示了它在包括偏微分方程、高斯过程、等变模型构建和无监督学习在内的广泛应用中的有效性。