GPU-parallelized MATLAB software for atom-ion dynamics
Description
We present a MATLAB script, atomiongpu.m, which can use GPU parallelization to run several million independent simulations per day of a trapped ion interacting with a low-density cloud of atoms, calculating classical 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, on a CPU, solve it about 7x faster than MATLAB’s ode45, leading to a 600x-3500x speedup when running a million trajectories using ode45gpu in parallel on a GPU compared to ode45 on a CPU. 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.