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.

17 pages, 12 figures

References in corpus (10)

• Machine learning for molecular simulation
• Jastrow correlation factor for atoms, molecules, and solids
• Restricted-Boltzmann-Machine Learning for Solving Strongly Correlated Quantum Systems
• Markov Chain Monte Carlo: Can We Trust the Third Significant Figure?
• A New Generation of Effective Core Potentials for Correlated Calculations
• Direct comparison of many-body methods for realistic electronic Hamiltonians
• Interatomic force from neural network based variational quantum Monte Carlo
• Better, Faster Fermionic Neural Networks
• Generalizing Neural Wave Functions
• Towards a Foundation Model for Neural Network Wavefunctions

Cited by in corpus (9)

• 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
• 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