NetKet: A Machine Learning Toolkit for Many-Body Quantum Systems

Abstract

We introduce NetKet, a comprehensive open source framework for the study of many-body quantum systems using machine learning techniques. The framework is built around a general and flexible implementation of neural-network quantum states, which are used as a variational ansatz for quantum wave functions. NetKet provides algorithms for several key tasks in quantum many-body physics and quantum technology, namely quantum state tomography, supervised learning from wave-function data, and ground state searches for a wide range of customizable lattice models. Our aim is to provide a common platform for open research and to stimulate the collaborative development of computational methods at the interface of machine learning and many-body physics.

References in corpus (21)

• The density-matrix renormalization group in the age of matrix product states
• Solving the Quantum Many-Body Problem with Artificial Neural Networks
• On the Measurement of Qubits
• Many-body quantum state tomography with neural networks
• The ALPS project release 1.3: open source software for strongly correlated systems
• The ALPS project release 2.0: Open source software for strongly correlated systems
• A class of ansatz wave functions for 1D spin systems and their relation to DMRG
• Neural-Network Approach to Dissipative Quantum Many-Body Dynamics
• Green Function Monte Carlo with Stochastic Reconfiguration
• Symmetries and many-body excited states with neural-network quantum states
• Variational Quantum Monte Carlo Method with a Neural-Network Ansatz for Open Quantum Systems
• Variational neural network ansatz for steady states in open quantum systems
• Approximating quantum many-body wave-functions using artificial neural networks
• Localization and Glassy Dynamics Of Many-Body Quantum Systems
• Geminal wavefunctions with Jastrow correlation: a first application to atoms
• Constructing neural stationary states for open quantum many-body systems
• Latent Space Purification via Neural Density Operators
• Solving the Bose-Hubbard model with machine learning
• Machine learning technique to find quantum many-body ground states of bosons on a lattice
• Chiral Topological Phases from Artificial Neural Networks
• Quenches near Ising quantum criticality as a challenge for artificial neural networks

Cited by in corpus (87)

• Fermionic neural-network states for ab-initio electronic structure
• Study of the Two-Dimensional Frustrated J1-J2 Model with Neural Network Quantum States
• NetKet 3: Machine Learning Toolbox for Many-Body Quantum Systems
• Review: Deep Learning in Electron Microscopy
• Machine learning quantum states in the NISQ era
• Restricted Boltzmann Machines for Quantum States with Nonabelian or Anyonic Symmetries
• Fermionic Wave Functions from Neural-Network Constrained Hidden States
• A Tutorial on Quantum Master Equations: Tips and tricks for quantum optics, quantum computing and beyond
• Neural network wave functions and the sign problem
• Precise measurement of quantum observables with neural-network estimators
• Transfer learning for scalability of neural-network quantum states
• Unbiasing time-dependent Variational Monte Carlo by projected quantum evolution
• Speeding up Learning Quantum States through Group Equivariant Convolutional Quantum Ansätze
• Solving Quasiparticle Band Spectra of Real Solids using Neural-Network Quantum States
• Message-Passing Neural Quantum States for the Homogeneous Electron Gas
• Eigenstate extraction with neural-network tomography
• Purifying Deep Boltzmann Machines for Thermal Quantum States
• Analyzing non-equilibrium quantum states through snapshots with artificial neural networks
• From Tensor Network Quantum States to Tensorial Recurrent Neural Networks
• Gauge equivariant neural networks for quantum lattice gauge theories
• Role of stochastic noise and generalization error in the time propagation of neural-network quantum states
• Scaling of neural-network quantum states for time evolution
• Self-learning projective quantum Monte Carlo simulations guided by restricted Boltzmann machines
• Many-Body Quantum States with Exact Conservation of Non-Abelian and Lattice Symmetries through Variational Monte Carlo
• Correlation-Enhanced Neural Networks as Interpretable Variational Quantum States
• Ab-initio variational wave functions for the time-dependent many-electron Schrödinger equation
• Phase diagram of the $J$-$J_d$ Heisenberg Model on the Maple-Leaf Lattice: Neural networks and density matrix renormalization group
• A Kaczmarz-inspired approach to accelerate the optimization of neural network wavefunctions
• Machine learning effective models for quantum systems
• Can neural quantum states learn volume-law ground states?
• Natural evolution strategies and variational Monte Carlo
• QuantumToolbox.jl: An efficient Julia framework for simulating open quantum systems
• jVMC: Versatile and performant variational Monte Carlo leveraging automated differentiation and GPU acceleration
• Neural Projected Quantum Dynamics: a systematic study
• Solving and visualizing fractional quantum Hall wavefunctions with neural network
• Is attention all you need to solve the correlated electron problem?
• Matrix Product States with Backflow correlations
• Spin-1/2 kagome Heisenberg antiferromagnet: Machine learning discovery of the spinon pair density wave ground state
• The quantum Gaussian process state: A kernel-inspired state with quantum support data
• A framework for efficient ab initio electronic structure with Gaussian Process States
• Real-time Dynamics of the Schwinger Model as an Open Quantum System with Neural Density Operators
• Learning ground states of gapped quantum Hamiltonians with Kernel Methods
• Scalable Imaginary Time Evolution with Neural Network Quantum States
• Many-Qudit representation for the Travelling Salesman Problem Optimisation
• Ground State Properties of Quantum Skyrmions described by Neural Network Quantum States
• Empirical Sample Complexity of Neural Network Mixed State Reconstruction
• Impact of conditional modelling for a universal autoregressive quantum state
• Phase diagram reconstruction of the Bose-Hubbard model with a Restricted Boltzmann Machine wavefunction
• Quantum walk versus classical wave: Distinguishing ground states of quantum magnets by spacetime dynamics
• Neural network enhanced hybrid quantum many-body dynamical distributions
• Lee-Yang theory of quantum phase transitions with neural network quantum states
• Machine learning of phases and structures for model systems in physics
• Neural Network Quantum States analysis of the Shastry-Sutherland model
• Towards quantum gravity with neural networks: Solving the quantum Hamilton constraint of U(1) BF theory
• Efficiency of neural quantum states in light of the quantum geometric tensor
• Predicting Topological Entanglement Entropy in a Rydberg analog simulator
• Restricted Boltzmann machine representation for the groundstate and excited states of Kitaev Honeycomb model
• Time-dependent Neural Galerkin Method for Quantum Dynamics
• Supervised Training of Neural-Network Quantum States for the Next Nearest Neighbor Ising model
• DeepQuark: A Deep-Neural-Network Approach to Multiquark Bound States
• Towards quantum gravity with neural networks: Solving quantum Hamilton constraints of 3d Euclidean gravity in the weak coupling limit
• Non-stabilizerness of Neural Quantum States
• Spectroscopy of two-dimensional interacting lattice electrons using symmetry-aware neural backflow transformations
• Liouville Space Neural Network Representation of Density Matrices
• Inferring Hidden Symmetries of Exotic Magnets from Detecting Explicit Order Parameters
• Broken-Symmetry Ground States of the Heisenberg model on the Pyrochlore Lattice
• Investigation of bi-particle states in gate-array-controlled quantum-dot systems aided by machine learning techniques
• Machine learning the deuteron: new architectures and uncertainty quantification
• Zero-temperature Monte Carlo simulations of two-dimensional quantum spin glasses guided by neural network states
• Design principles of deep translationally-symmetric neural quantum states for frustrated magnets
• Mott Transition and Volume Law Entanglement with Neural Quantum States
• Simple Fermionic backflow states via a systematically improvable tensor decomposition
• Efficiency of the hidden fermion determinant states Ansatz in the light of different complexity measures
• Phase diagram and crystal melting of helium-4 in two dimensions
• SU(d)-Symmetric Random Unitaries: Quantum Scrambling, Error Correction, and Machine Learning
• Machine-learning semi-local density functional theory for many-body lattice models at zero and finite temperature
• Scalable Effective Models for Superconducting Nanostructures: Applications to Double, Triple, and Quadruple Quantum Dots
• Neural Wave Functions for High-Pressure Atomic Hydrogen
• Universal Performance Gap of Neural Quantum States Applied to the Hofstadter-Bose-Hubbard Model
• Modal Backflow Neural Quantum States for Anharmonic Vibrational Calculations
• Optimizing Temperature Distributions for Training Neural Quantum States using Parallel Tempering
• Neural-Network Quantum States for Periodic Systems in Continuous Space
• Energy gap of quantum spin glasses: a projection quantum Monte Carlo study
• The Quantum Cartpole: A benchmark environment for non-linear reinforcement learning
• Group Convolutional Neural Network for the Low-Energy Spectrum in the Quantum Dimer Model
• Comparing Symmetrized Determinant Neural Quantum States for the Hubbard Model
• Learning a compass spin model with neural network quantum states