DeepAI AI Chat
Log In Sign Up

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

05/03/2019
by   Ashley Milsted, et al.
0

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.

READ FULL TEXT

page 1

page 2

page 3

page 4

06/07/2019

TensorNetwork for Machine Learning

We demonstrate the use of tensor networks for image classification with ...
05/03/2019

TensorNetwork: A Library for Physics and Machine Learning

TensorNetwork is an open source library for implementing tensor network ...
11/26/2020

ShapeFlow: Dynamic Shape Interpreter for TensorFlow

We present ShapeFlow, a dynamic abstract interpreter for TensorFlow whic...
03/07/2021

Implementing graph neural networks with TensorFlow-Keras

Graph neural networks are a versatile machine learning architecture that...
01/05/2018

Tensor Train decomposition on TensorFlow (T3F)

Tensor Train decomposition is used across many branches of machine learn...
10/20/2018

The Ocean Tensor Package

Matrix and tensor operations form the basis of a wide range of fields an...
04/27/2021

Active learning of tree tensor networks using optimal least-squares

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

Code Repositories

unitree

uniform tree tensor network using TensorNetwork


view repo