arXiv:2512.16677 · The Journal of Chemical Physics
QMCkl: A kernel library for quantum Monte Carlo applications
E. Slootman, Vijay Gopal Chilkuri, Aurélien Delval, Max Hoffer, Tommaso Gorni, François Coppens, Joris van de Nes, Ramón L. Panadés‐Barrueta, Evgeny Posenitskiy, Abdallah Ammar, Edgar Josué Landinez Borda, Kevin Camus, Oto Kohulák, Emmanuel Giner, Pablo de Oliveira Castro, Cédric Valensi, William Jalby, Claudia Filippi, Anthony Scemama
Quantum Monte Carlo (QMC) methods deliver highly accurate electronic structure calculations but are computationally intensive. The quantum Monte Carlo kernel library (QMCkl) provides a modular, portable collection of high-performance kernels implementing the core building blocks of QMC calculations. It offers a C-compatible application programming interface, supports the TREXIO standard for input, and covers essential QMC kernels including atomic and molecular orbitals, cusp corrections, the Jastrow factor, and the necessary derivatives also to perform variational and structural optimization. QMCkl separates algorithmic development from hardware-specific tuning by combining human-readable reference implementations with performance-optimized kernels that produce identical numerical results. The library enables consistent, efficient, and reproducible simulations across different QMC codes and architectures and achieves substantial speedups in the evaluation of the energy and its derivatives. Beyond QMC, QMCkl can accelerate deterministic quantum chemistry workflows and visualization tools, promoting cross-code interoperability and simplifying high-performance scientific software development.
Quantum Monte Carlo (QMC) methods deliver highly accurate electronic structure calculations but are computationally intensive. The quantum Monte Carlo kernel library (QMCkl) provides a modular, portable collection of high-performance kernels implementing the core building blocks of QMC calculations. It offers a C-compatible application programming interface, supports the TREXIO standard for input, and covers essential QMC kernels including atomic and molecular orbitals, cusp corrections, the Jastrow factor, and the necessary derivatives also to perform variational and structural optimization. QMCkl separates algorithmic development from hardware-specific tuning by combining human-readable reference implementations with performance-optimized kernels that produce identical numerical results. The library enables consistent, efficient, and reproducible simulations across different QMC codes and architectures and achieves substantial speedups in the evaluation of the energy and its derivatives. Beyond QMC, QMCkl can accelerate deterministic quantum chemistry workflows and visualization tools, promoting cross-code interoperability and simplifying high-performance scientific software development.
There are three different Quantum Monte Carlo (QMC) codes in the Targeting REal chemical accuracy at the eXascale (TREX) Center of Excellence (CoE), TurboRVB, Champ, and QMC=Chem, each with its own strengths and weaknesses. Instead of optimizing the three codes independently for exascale architectures, or re-writing a new monolithic code, our strategy is instead to design a new library, Quantum Monte Carlo kernel library (QMCkl), containing the state-of-the-art expertise in the implementation of specific kernels present in each of the three codes. The functions available in this library will allow all the codes to benefit from the optimal implementation of the major kernels of QMC. The three main objectives driving the development of QMCkl are performance, productivity and portability.
QMCPACK is an open source quantum Monte Carlo package for ab initio electronic structure calculations. It supports calculations of metallic and insulating solids, molecules, atoms, and some model Hamiltonians. Implemented real space quantum Monte Carlo algorithms include variational, diffusion, and reptation Monte Carlo. QMCPACK uses Slater-Jastrow type trial wavefunctions in conjunction with a sophisticated optimizer capable of optimizing tens of thousands of parameters. The orbital space auxiliary-field quantum Monte Carlo method is also implemented, enabling cross validation between different highly accurate methods. The code is specifically optimized for calculations with large numbers of electrons on the latest high performance computing architectures, including multicore central processing unit and graphical processing unit systems. We detail the program's capabilities, outline its structure, and give examples of its use in current research calculations. The package is available at http://qmcpack.org.
We present jQMC, a Python-based computational package for {\it ab initio} Quantum Monte Carlo (QMC) simulations, designed for modern GPU-accelerated computing environments. jQMC implements two well-established QMC algorithms: Variational Monte Carlo (VMC) and the lattice-regularized variant of Diffusion Monte Carlo (LRDMC). The employed wave function is a Jastrow factor combined with the antisymmetrized geminal power with spin-singlet and spin-triplet pairings, which contains the single Slater determinant as its special lowest-rank case. The wave function can be initialized from external Hartree-Fock/Density Functional Theory calculations through the TREX-IO library (a common wave-function format across electronic-structure packages) and optimized by stochastic reconfiguration and linear-method energy minimization. One of the prominent features of jQMC is its use of JAX, which enables automatic differentiation for wave function optimization and atomic force calculations, and allows the main QMC algorithms to be Just-In-Time (JIT) compiled and portable across CPU and GPU. jQMC is vectorized over walkers at the top level of the QMC algorithms, providing efficient intra-GPU~(CPU) vectorization. The multi-GPU~(CPU) parallelization is also supported through MPI and JAX sharding. To assess the practical performance of this implementation, we benchmarked jQMC performance on NVIDIA GPUs (A100 and H100) and analyzed CUDA kernels. For the test cases analyzed here, with system sizes up to 160 electrons, the current version of jQMC is faster than TurboRVB, a Fortran90 code implementing the same algorithms and wave functions, once jQMC is run on GPUs. In terms of wall-time, the gain can reach an order of magnitude for VMC, while it is more moderate for LRDMC.
TREX : an innovative view of HPC usage applied to Quantum Monte Carlo simulations The TREX[1] European Center of Excellence focuses on high accuracy quantum mechanical methods, essential in many different fields of application such as new material design or photochemistry. Among these methods, Quantum Monte Carlo (QMC) approaches are particularly well adapted to exascale architectures. Our ambition is to the help the community take advantage of exascale machines through the use of our HPC software. We will review in the presentation progress along the three following axes: * TREXIO[2]: A common I/O library and file format for easily exchanging data between applications, facilitating high-throughput computing workflows, * QMCkl[3]: A library of computational kernels specific to QMC applications written together by QMC and HPC experts, taking advantage of both CPUs and GPUs, * An integrated workflow including performance analysis (MAQAO[4]) and numerical accuracy measurements (Verificarlo[5]) to be used for the development of QMC kernels and more generally for improving the applications. In particular, we plan to identify the best performance usage of QMCkl and also to adjust the performance with numerical precision requirements. ------------------------------------ [1] https://trex-coe.eu [2] https://github.com/trex-coe/trexio [3] https://trex-coe.github.io/qmckl [4] https://www.maqao.org [5] https://github.com/verificarlo ------------------------------------
Quantum Monte Carlo (QMC) is one of the most promising methods for solving quantum many-body problems. QMC combines known analytical properties of wave functions, results from other methods, and stochastic techniques, into a powerful tool for investigation of the electronic structure of real materials. Furthermore, the method has a favorable scaling ( O[N3] in the number of electrons), perfect scalability on parallel architectures, and can be applied to a wide range of systems. Our work has focused both on expanding the capabilities of QMC for calculation of quantities beyond energies, such as interatomic forces, and on application of the method to challenging problems in nanostructure materials research. For the calculation of interatomic forces we have implemented a finite difference correlated sampling method. The correlated sampling enables us to avoid the problem of statistical noise in evaluating the energy differences and allows us to obtain forces with an average 1% error. We have analyzed the effect of the method's main approximations on the estimate of the forces, both by theoretical arguments and by performing a variety of tests on small systems. Furthermore, we have used this newly developed method for estimating equilibrium geometries in a problem where other methods have failed. QMC enabled us to treat the electron correlation effects with high accuracy and to provide accurate predictions in a number of interesting applications. In particular, for carbon rings of C4N+2 stoichiometry we have resolved a long-standing problem regarding the competition between the aromaticity and second order Jahn-Teller effects. We have found that the dominant mechanism at small and intermediate sizes, which stabilizes the bond angle and bond length alternated geometries, is the secondorder Jahn-Teller effect while the aromatic isomer is always found to be a transition state. These high accuracy calculations which have involved multi-reference wave functions, extensive variational optimizations, and the use of massively parallel platforms demonstrate clearly the unique capabilities of the QMC approach.
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 Ansätze 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 from both the quantum chemistry and the machine learning communities.