TensorNetwork on TensorFlow: A Spin Chain Application Using Tree Tensor Networks
Ashley Milsted, Martin Ganahl, Stefan Leichenauer, Jack Hidary, Guifre Vidal
Code Available — Be the first to reproduce this paper.
ReproduceCode
- github.com/google/TensorNetworkOfficialIn paperjax★ 0
- github.com/amilsted/unitreetf★ 0
- github.com/amilsted/unitreettf★ 0
Abstract
TensorNetwork is an open source library for implementing tensor network algorithms in TensorFlow. We describe a tree tensor network (TTN) algorithm for approximating the ground state of either a periodic quantum spin chain (1D) or a lattice model on a thin torus (2D), and implement the algorithm using TensorNetwork. We use a standard energy minimization procedure over a TTN ansatz with bond dimension , with a computational cost that scales as O(^4). Using bond dimension [32,256] we compare the use of CPUs with GPUs and observe significant computational speed-ups, up to a factor of 100, using a GPU and the TensorNetwork library.