GPU-Accelerated Simulations of Thermal Transport using Machine Learning Molecular Dynamics

Controlling thermal conductivities of materials is important for a wide range of applications, from thermoelectrics for clean energy generation to electronic devices and thermal barrier coatings. The thermal conductivity is commonly estimated using molecular dynamics simulations within the Green-Kubo formulation. This requires a force field that is both 1) an accurate estimate of the interatomic interactions and 2) fast enough to allow simulations with sufficiently large length and time scales. Traditionally, only empirical force fields have fulfilled both of these requirements, which severely limits the applicability of the method. <br/> <br/>In this work, we employ the Gaussian Process-based FLARE force field, which automatically learns the interactions of more complex materials than empirical force fields. The resulting model can then be mapped to a low-dimensional, computationally efficient model. Through GPU-acceleration with LAMMPS and the Kokkos library, we achieve excellent performance and obtain well-converged estimates for the thermal conductivity. Furthermore, we investigate state-of-the-art sampling and spectral denoising methods for further acceleration of the simulations.