GPU-Accelerated MATLAB Software for Atom-Ion Dynamics
Abstract
We present a MATLAB script which can use GPU acceleration to simulate a trapped ion interacting with a low-density cloud of atoms. This script, called atomiongpu.m, can massively parallelize MD simulations of trajectories of a trapped ion and an atom starting far away. The script uses ode45gpu, which is our optimized and specialized implementation of the Runge-Kutta algorithm used in MATLAB's ODE solver ode45. We first discuss the physical system and show how ode45gpu can solve it up to 22x faster than MATLAB's ode45. Then, we show how to easily modify the inputs to atomiongpu.m to account for different kinds of atoms, ions, atom-ion interactions, trap potentials, simulation parameters, initial conditions, and computational hardware, so that atomiongpu.m automatically finds the probability of complex formation, the distribution of observables such as the scattering angle and complex lifetime, and plots of specific trajectories.