DeepAI AI Chat
Log In Sign Up

TensorNetwork on TensorFlow: A Spin Chain Application Using Tree Tensor Networks

by   Ashley Milsted, et al.

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.


page 1

page 2

page 3

page 4


TensorNetwork for Machine Learning

We demonstrate the use of tensor networks for image classification with ...

TensorNetwork: A Library for Physics and Machine Learning

TensorNetwork is an open source library for implementing tensor network ...

ShapeFlow: Dynamic Shape Interpreter for TensorFlow

We present ShapeFlow, a dynamic abstract interpreter for TensorFlow whic...

Implementing graph neural networks with TensorFlow-Keras

Graph neural networks are a versatile machine learning architecture that...

Tensor Train decomposition on TensorFlow (T3F)

Tensor Train decomposition is used across many branches of machine learn...

The Ocean Tensor Package

Matrix and tensor operations form the basis of a wide range of fields an...

Active learning of tree tensor networks using optimal least-squares

In this paper, we propose new learning algorithms for approximating high...

Code Repositories


uniform tree tensor network using TensorNetwork

view repo