jcmgray/quimb

解决的问题

quimb 专为量子信息多体计算而设计。它简化了复杂量子模拟的过程，特别是通过提供用于张量网络和精确矩阵基量子计算的工具，这些计算通常计算成本高且难以手动管理。

工作原理

该库分为两个主要功能区域：

1. 张量网络（quimb.tensor）： 此模块允许用户创建和操作具有任意几何结构（包括超图）的张量网络。它可自动完成这些网络的收缩、优化和可视化，并通过 autoray 支持 JAX 和 PyTorch 等多种后端数组库。
2. 精确计算（核心 quimb）： 此模块处理以 NumPy 或 SciPy 稀疏矩阵表示状态和算符的参考计算。它提供寻找基态、激发态以及执行时间演化等工具，通常由 Numba 或 SLEPc 加速。

适用人群

适用于需要使用张量网络算法（如 DMRG、TEBD、MERA 或 PEPS）或精确对角化方法来模拟量子系统的量子信息科学和多体物理领域的研究人员和开发者。

主要亮点

• 灵活的几何结构： 支持超越标准一维和二维晶格的张量网络。
• 后端无关： 与 JAX 和 PyTorch 集成，实现硬件加速。
• 算法支持： 实现了特定的量子电路算法和张量网络方法，如 DMRG 和 TEBD。
• 性能工具： 利用 Numba 加速和 SLEPc 进行高级矩阵运算。