.png){kind=icon}
Return
Density Matrix Renormalization Group with Tensor Processing Units
.png){kind=icon}
OA
AI
DOI：10.1103/PRXQuantum.4.010317.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
Google's tensor processing units (TPUs) are integrated circuits specifically built to accelerate and scale up machine learning workloads. They can perform fast distributed matrix multiplications and therefore be repurposed for other computationally intensive tasks. In this work we demonstrate the use of TPUs for accelerating and scaling up the density matrix renormalization group (DMRG), a powerful numerical approach to compute the ground state of a local quantum many-body Hamiltonian. The cost of DMRG scales with system size N as O(ND3), where the so-called bond dimension D regulates how expressive the underlying matrix product state (MPS) variational ansatz is. We consider lattice models in two spatial dimensions, with square lattices of size 10 x 10 (free fermions) and 20 x 20 (transverse field Ising model), for which the required MPS bond dimension is known to scale at least as exp(root N). Using half of a TPU v3 pod (namely 1024 TPU v3 cores), we reach an unprecedentedly large bond dimension D = 216 = 65536, for which optimizing a single MPS tensor takes about 2 min.
Keywords:
ENTANGLEMENT
ALGORITHM
STATES
ENTROPY
SYSTEMS
PHASE
MODEL
Journal
P
IF:
11
Papers:
919
Citations:
9.0K
Organization
Cited Papers
ShareSaveA six-coordinated cationic ruthenium carbyne complex with liable pyridine ligands: synthesis, structure, catalytic investigation, and DFT study on initiation mechanism.png)
Tetrahedron
TetrahedronIF0
PREAI
ShareSaveA ratiometric, fluorescent BODIPY-based probe for transition and heavy metal ionsRSC AdvancesOAAI
RSC AdvancesIF0
OAAIShareSaveShareSaveSpectral functions in one-dimensional quantum systems at finite temperature using the density matrix renormalization groupPHYSICAL REVIEW B
OAAI
PHYSICAL REVIEW BIF3.7
OAAIShareSaveShareSave
View more