NetKet 3: Machine Learning Toolbox for Many-Body Quantum Systems

Abstract

We introduce version 3 of NetKet, the machine learning toolbox for many-body quantum physics. NetKet is built around neural-network quantum states and provides efficient algorithms for their evaluation and optimization. This new version is built on top of JAX, a differentiable programming and accelerated linear algebra framework for the Python programming language. The most significant new feature is the possibility to define arbitrary neural network ansätze in pure Python code using the concise notation of machine-learning frameworks, which allows for just-in-time compilation as well as the implicit generation of gradients thanks to automatic differentiation. NetKet 3 also comes with support for GPU and TPU accelerators, advanced support for discrete symmetry groups, chunking to scale up to thousands of degrees of freedom, drivers for quantum dynamics applications, and improved modularity, allowing users to use only parts of the toolbox as a foundation for their own code.

55 pages, 5 figures. Accompanying code at https://github.com/netket/netket

References in corpus (8)

• QuTiP 2: A Python framework for the dynamics of open quantum systems
• Neural-Network Approach to Dissipative Quantum Many-Body Dynamics
• 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
• Variational principle for steady states of dissipative quantum many-body systems
• Gapless spin liquid and valence-bond solid in the $J_1-J_2$ Heisenberg model on the square lattice: insights from singlet and triplet excitations
• Investigating ultrafast quantum magnetism with machine learning
• Neural-Network Quantum States for Periodic Systems in Continuous Space

Cited by in corpus (87)

• Beyond-classical computation in quantum simulation
• A Tutorial on Quantum Master Equations: Tips and tricks for quantum optics, quantum computing and beyond
• Empowering deep neural quantum states through efficient optimization
• Variational Benchmarks for Quantum Many-Body Problems
• High-accuracy variational Monte Carlo for frustrated magnets with deep neural networks
• A simple linear algebra identity to optimize Large-Scale Neural Network Quantum States
• Unbiasing time-dependent Variational Monte Carlo by projected quantum evolution
• Message-Passing Neural Quantum States for the Homogeneous Electron Gas
• Imaginary components of out-of-time correlators and information scrambling for navigating the learning landscape of a quantum machine learning model
• From Tensor Network Quantum States to Tensorial Recurrent Neural Networks
• Dynamics with autoregressive neural quantum states: application to critical quench dynamics
• Transformer Wave Function for two dimensional frustrated magnets: emergence of a Spin-Liquid Phase in the Shastry-Sutherland Model
• 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
• Variational Neural-Network Ansatz for Continuum Quantum Field Theory
• Variational solutions to fermion-to-qubit mappings in two spatial dimensions
• QuantumToolbox.jl: An efficient Julia framework for simulating open quantum systems
• Foundation Neural-Networks Quantum States as a Unified Ansatz for Multiple Hamiltonians
• Can neural quantum states learn volume-law ground states?
• Artificial Intelligence for Science in Quantum, Atomistic, and Continuum Systems
• $m^{\ast }$ of two-dimensional electron gas: a neural canonical transformation study
• Neural Projected Quantum Dynamics: a systematic study
• $2^{1296}$ Exponentially Complex Quantum Many-Body Simulation via Scalable Deep Learning Method
• Ground state search by local and sequential updates of neural network quantum states
• Matrix Product States with Backflow correlations
• Spin-1/2 kagome Heisenberg antiferromagnet: Machine learning discovery of the spinon pair density wave ground state
• Comment on "Can Neural Quantum States Learn Volume-Law Ground States?"
• Neural-Shadow Quantum State Tomography
• A framework for efficient ab initio electronic structure with Gaussian Process States
• Learning ground states of gapped quantum Hamiltonians with Kernel Methods
• Real-time Dynamics of the Schwinger Model as an Open Quantum System with Neural Density Operators
• The percolating cluster is invisible to image recognition with deep learning
• Scalable Imaginary Time Evolution with Neural Network Quantum States
• Fine-tuning Neural Network Quantum States
• Are queries and keys always relevant? A case study on Transformer wave functions
• Ground State Properties of Quantum Skyrmions described by Neural Network Quantum States
• Impact of conditional modelling for a universal autoregressive quantum state
• Deep learning lattice gauge theories
• Accurate neural quantum states for interacting lattice bosons
• Empirical Sample Complexity of Neural Network Mixed State Reconstruction
• Deep Neural Networks as Variational Solutions for Correlated Open Quantum Systems
• Hybrid Ground-State Quantum Algorithms based on Neural Schrödinger Forging
• Transformer Wave Function for Quantum Long-Range models
• Lee-Yang theory of quantum phase transitions with neural network quantum states
• Neural Network Quantum States analysis of the Shastry-Sutherland model
• Predicting Topological Entanglement Entropy in a Rydberg analog simulator
• Approximately-symmetric neural networks for quantum spin liquids
• Efficiency of neural quantum states in light of the quantum geometric tensor
• Second-order optimisation strategies for neural network quantum states
• Spectroscopy of two-dimensional interacting lattice electrons using symmetry-aware neural backflow transformations
• DeepQuark: A Deep-Neural-Network Approach to Multiquark Bound States
• Non-stabilizerness of Neural Quantum States
• Time-dependent Neural Galerkin Method for Quantum Dynamics
• Machine learning one-dimensional spinless trapped fermionic systems with neural-network quantum states
• SOLAX: A Python solver for fermionic quantum systems with neural network support
• Variational Neural and Tensor Network Approximations of Thermal States
• Synergy between deep neural networks and the variational Monte Carlo method for small ${}^{4}H{e}_N$ clusters
• Machine learning the deuteron: new architectures and uncertainty quantification
• Neural Quantum State Study of Fracton Models
• Noncoplanar and chiral spin states on the way towards Néel ordering in fullerene Heisenberg models
• Design principles of deep translationally-symmetric neural quantum states for frustrated magnets
• Mott Transition and Volume Law Entanglement with Neural Quantum States
• Determinant- and Derivative-Free Quantum Monte Carlo Within the Stochastic Representation of Wavefunctions
• 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
• Partial suppression of magnetism in the square lattice SU(3) Hubbard model
• Local fermion-to-qudit mappings: a practical recipe for four-level systems
• Variational Transformer Ansatz for the Density Operator of Steady States in Dissipative Quantum Many-Body Systems
• Correcting and extending Trotterized quantum many-body dynamics
• Neural Wave Functions for High-Pressure Atomic Hydrogen
• Phase diagram of the J1-J2 Heisenberg second-order topological quantum magnet
• Finding the Dynamics of an Integrable Quantum Many-Body System via Machine Learning
• Scalable Effective Models for Superconducting Nanostructures: Applications to Double, Triple, and Quadruple Quantum Dots
• An Empirical Study of Quantum Dynamics as a Ground State Problem with Neural Quantum States
• Universal Performance Gap of Neural Quantum States Applied to the Hofstadter-Bose-Hubbard Model
• Modal Backflow Neural Quantum States for Anharmonic Vibrational Calculations
• Basis dependence of Neural Quantum States for the Transverse Field Ising Model
• Energy gap of quantum spin glasses: a projection quantum Monte Carlo study
• Variational decision diagrams for quantum-inspired machine learning applications
• Comparing Symmetrized Determinant Neural Quantum States for the Hubbard Model
• The toric code under antiferromagnetic isotropic Heisenberg interactions
• Optimizing Temperature Distributions for Training Neural Quantum States using Parallel Tempering
• Majorana string simulation of nonequilibrium dynamics in two-dimensional lattice fermion systems
• Group Convolutional Neural Network for the Low-Energy Spectrum in the Quantum Dimer Model
• Interpretable representation learning of quantum data enabled by probabilistic variational autoencoders
• Magnetic correlations in the $SU(3)$ triangular-lattice $t$-$J$ model at finite doping