GitHub repository: stephenhky/TorchMPS
Description: PyTorch toolbox for matrix product state models