DeepQMC: an open-source software suite for variational optimization of deep-learning molecular wave functions

Abstract

Computing accurate yet efficient approximations to the solutions of the electronic Schrödinger equation has been a paramount challenge of computational chemistry for decades. Quantum Monte Carlo methods are a promising avenue of development as their core algorithm exhibits a number of favorable properties: it is highly parallel, and scales favorably with the considered system size, with an accuracy that is limited only by the choice of the wave function ansatz. The recently introduced machine-learned parametrizations of quantum Monte Carlo ansatzes rely on the efficiency of neural networks as universal function approximators to achieve state of the art accuracy on a variety of molecular systems. With interest in the field growing rapidly, there is a clear need for easy to use, modular, and extendable software libraries facilitating the development and adoption of this new class of methods. In this contribution, the DeepQMC program package is introduced, in an attempt to provide a common framework for future investigations by unifying many of the currently available deep-learning quantum Monte Carlo architectures. Furthermore, the manuscript provides a brief introduction to the methodology of variational quantum Monte Carlo in real space, highlights some technical challenges of optimizing neural network wave functions, and presents example black-box applications of the program package. We thereby intend to make this novel field accessible to a broader class of practitioners both from the quantum chemistry as well as the machine learning communities.

arXiv comment17 pages, 12 figures

References in corpus (29)

• Decoupled Weight Decay Regularization
• Solving the Quantum Many-Body Problem with Artificial Neural Networks
• Recent developments in the PySCF program package
• Machine learning for molecular simulation
• Deep neural network solution of the electronic Schrödinger equation
• Ab-Initio Solution of the Many-Electron Schrödinger Equation with Deep Neural Networks
• Jastrow correlation factor for atoms, molecules, and solids
• Generalized Lanczos Algorithm for Variational Quantum Monte Carlo
• Restricted-Boltzmann-Machine Learning for Solving Strongly Correlated Quantum Systems
• Markov Chain Monte Carlo: Can We Trust the Third Significant Figure?
• Green Function Monte Carlo with Stochastic Reconfiguration
• Solving Many-Electron Schrödinger Equation Using Deep Neural Networks
• Ab-initio quantum chemistry with neural-network wavefunctions
• A New Generation of Effective Core Potentials for Correlated Calculations
• New generation of effective core potentials from correlated calculations: 3d transition metal series
• Direct comparison of many-body methods for realistic electronic Hamiltonians
• Benchmark all-electron ab initio quantum Monte Carlo calculations for small molecules
• Towards the ground state of molecules via diffusion Monte Carlo on neural networks
• Wave function Ansatz (but Periodic) Networks and the Homogeneous Electron Gas
• Determinant-free fermionic wave function using feed-forward neural networks
• A Self-Attention Ansatz for Ab-initio Quantum Chemistry
• Convergence to the fixed-node limit in deep variational Monte Carlo
• Interatomic force from neural network based variational quantum Monte Carlo
• Better, Faster Fermionic Neural Networks
• Gold-standard solutions to the Schrödinger equation using deep learning: How much physics do we need?
• Ab-Initio Potential Energy Surfaces by Pairing GNNs with Neural Wave Functions
• Sampling-free Inference for Ab-Initio Potential Energy Surface Networks
• Towards a Foundation Model for Neural Network Wavefunctions
• Generalizing Neural Wave Functions

Cited by in corpus (11)

• A Kaczmarz-inspired approach to accelerate the optimization of neural network wavefunctions
• Exploring the design space of machine-learning models for quantum chemistry with a fully differentiable framework
• Artificial-intelligence-based surrogate solution of dissipative quantum dynamics: physics-informed reconstruction of the universal propagator
• Highly Accurate Real-space Electron Densities with Neural Networks
• An improved penalty-based excited-state variational Monte Carlo approach with deep-learning ansatzes
• Neural Network Quantum States analysis of the Shastry-Sutherland model
• Guest Editorial: Special Topic on Software for Atomistic Machine Learning
• Deep quantum Monte Carlo approach for polaritonic chemistry
• Determinant- and Derivative-Free Quantum Monte Carlo Within the Stochastic Representation of Wavefunctions
• Expressivity of determinantal ansatzes for neural network wave functions
• Gaussian Processes for Finite Size Extrapolation of Many-Body Simulations