张量计算接口:一种面向应用、轻量级的便携式高性能张量网络应用程序接口
张量网络(TN)是量子科学与人工智能领域的核心计算工具。然而,由于张量计算框架间缺乏统一软件接口,TN应用的可移植性受到严重制约,导致算法开发与特定硬件及软件后端深度耦合。为应对这一挑战,该研究团队推出“张量计算接口”(TCI)——一种面向应用的轻量级应用程序接口,旨在实现与框架无关的高性能TN应用。TCI通过明确定义的类型系统抽象张量对象,并提供一组精简而富有表现力的核心函数,涵盖基本张量操作与张量线性代数运算。通过在典型张量网络应用中的数值实验,该工作表明基于TCI编写的代码可在异构硬件和软件平台间无缝迁移,同时获得与原生框架实现相当的性能表现。研究人员还发布了基于《Cytnx》的TCI开源实现,验证了其实际应用价值及与现有张量计算框架的便捷集成能力。
量科快讯
20 小时前
22 小时前
1 天前
1 天前
2 天前

