Johnnie Gray
Current Organizations / Institutes
The organizations/institutes from the scholar's latest article
Previous Organizations / Institutes
The organizations / institutes from the scholar's other articles in the past ten years
Research Subjects
Research Interests
Johnnie Gray has a total of 28 co-authors. Here are the top co-authors by number of shared publications:
- Garnet Kin-Lic Chan
California Institute of Technology
6 shared publications Garnet Kin-Lic Chan
California Institute of Technology
4 shared publicationsAbolfazl Bayat
Department of Physics and Astronomy, University College London, Gower Street, London WC1E 6BT, United Kingdom
3 shared publications- Gunhee Park
California Institute of Technology
3 shared publications - Ruojing Peng
California Institute of Technology
3 shared publications Sougato Bose
Department of Physics and Astronomy, University College London, Gower Street, London WC1E 6BT, United Kingdom
3 shared publications
Latest Publications
- 22 June 2026
- Published by American Physical Society (APS) in Physical Review B
- Vol. 113 (23)
Tensor network loop cluster expansions for quantum many-body problems
- 22 June 2026
- Published by American Physical Society (APS) in Physical Review B
- Vol. 113 (23)
Tensor network loop cluster expansions for quantum many-body problems
- Open Access
- 5 March 2026
- Published by American Physical Society (APS) in Physical Review Research
- Vol. 8 (1)
Loop series expansions for tensor networks
Belief propagation (BP) can be a useful tool to approximately contract a tensor network, provided that the contributions from any closed loops in the network are sufficiently weak. In this article, we describe how a loop series expansion can be applied to systematically improve the accuracy of a BP approximation to a tensor network contraction, in principle converging arbitrarily close to the exact result. More generally, our result provides a framework for expanding a tensor network as a sum of component networks in a hierarchy of increasing complexity. We benchmark this proposal for the contraction of infinite projected entangled pair states, either representing the ground state of an Affleck-Kennedy-Lieb-Tasaki (AKLT) model or with randomly defined tensors, where it is shown to improve in accuracy over standard BP by several orders of magnitude while incurring only a minor increase in computational cost. These results indicate that the proposed series expansions could be a useful tool to accurately evaluate tensor networks in cases that otherwise exceed the limits of established contraction routines.
- 5 March 2026
- Published by American Physical Society (APS) in Physical Review Research
- Vol. 8 (1)
Open AccessLoop series expansions for tensor networks
Belief propagation (BP) can be a useful tool to approximately contract a tensor network, provided that the contributions from any closed loops in the network are sufficiently weak. In this article, we describe how a loop series expansion can be applied to systematically improve the accuracy of a BP approximation to a tensor network contraction, in principle converging arbitrarily close to the exact result. More generally, our result provides a framework for expanding a tensor network as a sum of component networks in a hierarchy of increasing complexity. We benchmark this proposal for the contraction of infinite projected entangled pair states, either representing the ground state of an Affleck-Kennedy-Lieb-Tasaki (AKLT) model or with randomly defined tensors, where it is shown to improve in accuracy over standard BP by several orders of magnitude while incurring only a minor increase in computational cost. These results indicate that the proposed series expansions could be a useful tool to accurately evaluate tensor networks in cases that otherwise exceed the limits of established contraction routines.
- Top Cited Papers
- 17 November 2025
- Published by American Physical Society (APS) in Physical Review B
- Vol. 112 (17)
Simulating quantum dynamics in two-dimensional lattices with tensor network influence functional belief propagation
- 17 November 2025
- Published by American Physical Society (APS) in Physical Review B
- Vol. 112 (17)
Top Cited PapersSimulating quantum dynamics in two-dimensional lattices with tensor network influence functional belief propagation
- Open Access
- 27 May 2025
- Published by American Physical Society (APS) in Physical Review Research
- Vol. 7 (2)
Fermionic tensor network contraction for arbitrary geometries
We describe our implementation of fermionic tensor network contraction on arbitrary lattices within both a globally ordered and a locally ordered formalism. We provide a pedagogical description of these two conventions as implemented for the quimb library. Using hyperoptimized approximate contraction strategies, we present benchmark fermionic projected entangled pair state simulations of finite Hubbard models defined on the three-dimensional diamond lattice and random regular graphs. Published by the American Physical Society 2025
- 27 May 2025
- Published by American Physical Society (APS) in Physical Review Research
- Vol. 7 (2)
Open AccessFermionic tensor network contraction for arbitrary geometries
We describe our implementation of fermionic tensor network contraction on arbitrary lattices within both a globally ordered and a locally ordered formalism. We provide a pedagogical description of these two conventions as implemented for the quimb library. Using hyperoptimized approximate contraction strategies, we present benchmark fermionic projected entangled pair state simulations of finite Hubbard models defined on the three-dimensional diamond lattice and random regular graphs. Published by the American Physical Society 2025
- 30 December 2024
- Published by American Physical Society (APS) in Physical Review Letters
- Vol. 133 (26) , 260404
Tensor Network Computations That Capture Strict Variationality, Volume Law Behavior, and the Efficient Representation of Neural Network States
We introduce a change of perspective on tensor network states that is defined by the computational graph of the contraction of an amplitude. The resulting class of states, which we refer to as tensor network functions, inherit the conceptual advantages of tensor network states while removing computational restrictions arising from the need to converge approximate contractions. We use tensor network functions to compute strict variational estimates of the energy on loopy graphs, analyze their expressive power for ground states, show that we can capture aspects of volume law time evolution, and provide a mapping of general feed-forward neural nets onto efficient tensor network functions. Our work expands the realm of computable tensor networks to ones where accurate contraction methods are not available, and opens up new avenues to use tensor networks.
- 30 December 2024
- Published by American Physical Society (APS) in Physical Review Letters
- Vol. 133 (26) , 260404
Tensor Network Computations That Capture Strict Variationality, Volume Law Behavior, and the Efficient Representation of Neural Network States
We introduce a change of perspective on tensor network states that is defined by the computational graph of the contraction of an amplitude. The resulting class of states, which we refer to as tensor network functions, inherit the conceptual advantages of tensor network states while removing computational restrictions arising from the need to converge approximate contractions. We use tensor network functions to compute strict variational estimates of the energy on loopy graphs, analyze their expressive power for ground states, show that we can capture aspects of volume law time evolution, and provide a mapping of general feed-forward neural nets onto efficient tensor network functions. Our work expands the realm of computable tensor networks to ones where accurate contraction methods are not available, and opens up new avenues to use tensor networks.
- 23 December 2024
- Published by American Physical Society (APS) in Physical Review E
- Vol. 110 (6) , 065306
Hyperoptimized approximate contraction of tensor networks for rugged-energy-landscape spin glasses on periodic square and cubic lattices
Obtaining the low-energy configurations of spin glasses that have rugged energy landscapes is of direct relevance to combinatorial optimization and fundamental science. Search-based heuristics have difficulty with this task due to the existence of many local minima that are far from optimal. The work of M. M. Rams et al. [Phys. Rev. E 104, 025308 (2021)2470-004510.1103/PhysRevE.104.025308] demonstrates an alternative that can bypass this issue for spin glasses with planar or quasiplanar geometry: sampling the Boltzmann distribution via approximate contractions of tensor networks. The computational complexity of this approach is due only to the complexity of contracting the network and is therefore independent of landscape ruggedness. Here we initiate an investigation of how to take this approach beyond (quasi-)planar geometry by utilizing hyperoptimized approximate contraction of tensor networks [J. Gray and G. K.-L. Chan, Phys. Rev. X 14, 011009 (2024)2160-330810.1103/PhysRevX.14.011009]. We perform tests on the periodic square- and cubic-lattice, planted-solution Ising spin glasses generated with tile planting [F. Hamze et al., Phys. Rev. E 97, 043303 (2018)2470-004510.1103/PhysRevE.97.043303] for up to 2304 (square lattice) and 216 (cubic lattice) spins. For a fixed bond dimension, the time complexity is quadratic. With a bond dimension of only four, over the tested system sizes, the average relative energy error in the most rugged instance class remains ∼1% (square lattice) or ∼10% (cubic lattice) of optimal. In less rugged instances the solution is always optimal for the square lattice and either optimal or within ∼1% for the cubic lattice. These results suggest that further development of optimization methods based on tensor-network representations of spin glass partition functions may be fruitful, especially given that such methods are not limited to the Ising (i.e., binary) or two-body (i.e., quadratic) settings.
- 23 December 2024
- Published by American Physical Society (APS) in Physical Review E
- Vol. 110 (6) , 065306
Hyperoptimized approximate contraction of tensor networks for rugged-energy-landscape spin glasses on periodic square and cubic lattices
Obtaining the low-energy configurations of spin glasses that have rugged energy landscapes is of direct relevance to combinatorial optimization and fundamental science. Search-based heuristics have difficulty with this task due to the existence of many local minima that are far from optimal. The work of M. M. Rams et al. [Phys. Rev. E 104, 025308 (2021)2470-004510.1103/PhysRevE.104.025308] demonstrates an alternative that can bypass this issue for spin glasses with planar or quasiplanar geometry: sampling the Boltzmann distribution via approximate contractions of tensor networks. The computational complexity of this approach is due only to the complexity of contracting the network and is therefore independent of landscape ruggedness. Here we initiate an investigation of how to take this approach beyond (quasi-)planar geometry by utilizing hyperoptimized approximate contraction of tensor networks [J. Gray and G. K.-L. Chan, Phys. Rev. X 14, 011009 (2024)2160-330810.1103/PhysRevX.14.011009]. We perform tests on the periodic square- and cubic-lattice, planted-solution Ising spin glasses generated with tile planting [F. Hamze et al., Phys. Rev. E 97, 043303 (2018)2470-004510.1103/PhysRevE.97.043303] for up to 2304 (square lattice) and 216 (cubic lattice) spins. For a fixed bond dimension, the time complexity is quadratic. With a bond dimension of only four, over the tested system sizes, the average relative energy error in the most rugged instance class remains ∼1% (square lattice) or ∼10% (cubic lattice) of optimal. In less rugged instances the solution is always optimal for the square lattice and either optimal or within ∼1% for the cubic lattice. These results suggest that further development of optimization methods based on tensor-network representations of spin glass partition functions may be fruitful, especially given that such methods are not limited to the Ising (i.e., binary) or two-body (i.e., quadratic) settings.
- Top Cited PapersOpen Access
- 26 January 2024
- Published by American Physical Society (APS) in Physical Review X
- Vol. 14 (1) , 011009
Hyperoptimized Approximate Contraction of Tensor Networks with Arbitrary Geometry
Tensor network contraction is central to problems ranging from many-body physics to computer science. We describe how to approximate tensor network contraction through bond compression on arbitrary graphs. In particular, we introduce a hyperoptimization over the compression and contraction strategy itself to minimize error and cost. We demonstrate that our protocol outperforms both handcrafted contraction strategies in the literature as well as recently proposed general contraction algorithms on a variety of synthetic and physical problems on regular lattices and random regular graphs. We further showcase the power of the approach by demonstrating approximate contraction of tensor networks for frustrated three-dimensional lattice partition functions, dimer counting on random regular graphs, and to access the hardness transition of random tensor network models, in graphs with many thousands of tensors. Published by the American Physical Society 2024
- 26 January 2024
- Published by American Physical Society (APS) in Physical Review X
- Vol. 14 (1) , 011009
Top Cited PapersOpen AccessHyperoptimized Approximate Contraction of Tensor Networks with Arbitrary Geometry
Tensor network contraction is central to problems ranging from many-body physics to computer science. We describe how to approximate tensor network contraction through bond compression on arbitrary graphs. In particular, we introduce a hyperoptimization over the compression and contraction strategy itself to minimize error and cost. We demonstrate that our protocol outperforms both handcrafted contraction strategies in the literature as well as recently proposed general contraction algorithms on a variety of synthetic and physical problems on regular lattices and random regular graphs. We further showcase the power of the approach by demonstrating approximate contraction of tensor networks for frustrated three-dimensional lattice partition functions, dimer counting on random regular graphs, and to access the hardness transition of random tensor network models, in graphs with many thousands of tensors. Published by the American Physical Society 2024
- Top Cited PapersOpen Access
- 19 January 2024
- Published by American Association for the Advancement of Science (AAAS) in Science Advances
- Vol. 10 (3) , eadk4321
Fast and converged classical simulations of evidence for the utility of quantum computing before fault tolerance
A recent quantum simulation of observables of the kicked Ising model on 127 qubits implemented circuits that exceed the capabilities of exact classical simulation. We show that several approximate classical methods, based on sparse Pauli dynamics and tensor network algorithms, can simulate these observables orders of magnitude faster than the quantum experiment and can also be systematically converged beyond the experimental accuracy. Our most accurate technique combines a mixed Schrödinger and Heisenberg tensor network representation with the Bethe free entropy relation of belief propagation to compute expectation values with an effective wave function–operator sandwich bond dimension >16,000,000, achieving an absolute accuracy, without extrapolation, in the observables of <0.01, which is converged for many practical purposes. We thereby identify inaccuracies in the experimental extrapolations and suggest how future experiments can be implemented to increase the classical hardness.
- 19 January 2024
- Published by American Association for the Advancement of Science (AAAS) in Science Advances
- Vol. 10 (3) , eadk4321
Top Cited PapersOpen AccessFast and converged classical simulations of evidence for the utility of quantum computing before fault tolerance
A recent quantum simulation of observables of the kicked Ising model on 127 qubits implemented circuits that exceed the capabilities of exact classical simulation. We show that several approximate classical methods, based on sparse Pauli dynamics and tensor network algorithms, can simulate these observables orders of magnitude faster than the quantum experiment and can also be systematically converged beyond the experimental accuracy. Our most accurate technique combines a mixed Schrödinger and Heisenberg tensor network representation with the Bethe free entropy relation of belief propagation to compute expectation values with an effective wave function–operator sandwich bond dimension >16,000,000, achieving an absolute accuracy, without extrapolation, in the observables of <0.01, which is converged for many practical purposes. We thereby identify inaccuracies in the experimental extrapolations and suggest how future experiments can be implemented to increase the classical hardness.
- Open Access
- 28 February 2023
- Published by American Physical Society (APS) in Physical Review Research
- Vol. 5 (1) , 013156
Arithmetic circuit tensor networks, multivariable function representation, and high-dimensional integration
Many computational problems can be formulated in terms of high-dimensional functions. Simple representations of such functions and resulting computations with them typically suffer from the “curse of dimensionality,” an exponential cost dependence on dimension. Tensor networks provide a way to represent certain classes of high-dimensional functions with polynomial memory. This results in computations where the exponential cost is ameliorated or, in some cases, removed, if the tensor network representation can be obtained. Here, we introduce a direct mapping from the arithmetic circuit of a function to arithmetic circuit tensor networks, avoiding the need to perform any optimization or functional fit. We demonstrate the power of the circuit construction in examples of multivariable integration on the unit hypercube in up to 50 dimensions, where the complexity of integration can be understood from the circuit structure. We find very favorable cost scaling compared with quasi–Monte Carlo integration for these cases and further give an example where efficient quasi–Monte Carlo integration cannot be performed without knowledge of the underlying tensor network circuit structure.
- 28 February 2023
- Published by American Physical Society (APS) in Physical Review Research
- Vol. 5 (1) , 013156
Open AccessArithmetic circuit tensor networks, multivariable function representation, and high-dimensional integration
Many computational problems can be formulated in terms of high-dimensional functions. Simple representations of such functions and resulting computations with them typically suffer from the “curse of dimensionality,” an exponential cost dependence on dimension. Tensor networks provide a way to represent certain classes of high-dimensional functions with polynomial memory. This results in computations where the exponential cost is ameliorated or, in some cases, removed, if the tensor network representation can be obtained. Here, we introduce a direct mapping from the arithmetic circuit of a function to arithmetic circuit tensor networks, avoiding the need to perform any optimization or functional fit. We demonstrate the power of the circuit construction in examples of multivariable integration on the unit hypercube in up to 50 dimensions, where the complexity of integration can be understood from the circuit structure. We find very favorable cost scaling compared with quasi–Monte Carlo integration for these cases and further give an example where efficient quasi–Monte Carlo integration cannot be performed without knowledge of the underlying tensor network circuit structure.
- Open Access
Arithmetic circuit tensor networks, multivariable function representation, and high-dimensional integration
Many computational problems can be formulated in terms of high-dimensional functions. Simple representations of such functions and resulting computations with them typically suffer from the "curse of dimensionality", an exponential cost dependence on dimension. Tensor networks provide a way to represent certain classes of high-dimensional functions with polynomial memory. This results in computations where the exponential cost is ameliorated or in some cases, removed, if the tensor network representation can be obtained. Here, we introduce a direct mapping from the arithmetic circuit of a function to arithmetic circuit tensor networks, avoiding the need to perform any optimization or functional fit. We demonstrate the power of the circuit construction in examples of multivariable integration on the unit hypercube in up to 50 dimensions, where the complexity of integration can be understood from the circuit structure. We find very favorable cost scaling compared to quasi-Monte-Carlo integration for these cases, and further give an example where efficient quasi-Monte-Carlo cannot be theoretically performed without knowledge of the underlying tensor network circuit structure.
Open AccessArithmetic circuit tensor networks, multivariable function representation, and high-dimensional integration
Many computational problems can be formulated in terms of high-dimensional functions. Simple representations of such functions and resulting computations with them typically suffer from the "curse of dimensionality", an exponential cost dependence on dimension. Tensor networks provide a way to represent certain classes of high-dimensional functions with polynomial memory. This results in computations where the exponential cost is ameliorated or in some cases, removed, if the tensor network representation can be obtained. Here, we introduce a direct mapping from the arithmetic circuit of a function to arithmetic circuit tensor networks, avoiding the need to perform any optimization or functional fit. We demonstrate the power of the circuit construction in examples of multivariable integration on the unit hypercube in up to 50 dimensions, where the complexity of integration can be understood from the circuit structure. We find very favorable cost scaling compared to quasi-Monte-Carlo integration for these cases, and further give an example where efficient quasi-Monte-Carlo cannot be theoretically performed without knowledge of the underlying tensor network circuit structure.