peps-torch
peps-torch is a python library for calculation of quantum many-body problems on two dimensional lattices. Variational principles calculation is used with an infinite PEPS (iPEPS) as the trial wave function. Therefore, the ground state is obtained in the form of the element tensor of the iPEPS. The energy of the trial state is estimated by the corner transfer matrix method (CTM), and its gradient with respect to the element tensor is computed through automatic differentiation provided by pytorch. Functions/classes for exploiting the system’s symmetry are provided for reducing the computational cost if possible. While general models and lattices are not supported, many examples of stand-alone codes would make it relatively easy for users to write their own codes to suit their needs. pytorch is required.