Abstract
Potentials derived with machine learning algorithms achieve the accuracy of high-fidelity quantum mechanical computations such as density functional theory (DFT), while allowing orders of magnitude lower computational time. In this work, we demonstrate the use of uncertainty aware equivariant graph neural networks for predicting spin-resolved electron densities, forces, and energies of the $Na_3V_2(PO_4)_3$ NASICON structured cathode. Due to the speedup in computational time, we are able to investigate structures of $\sim 300$ atoms for ~200 million timesteps. The ability to model larger systems on the nanosecond length scale with maintaining DFT level accuracy allowed critical insights into the diffusion characteristics of Na-ions, associated electron transfer processes, and dependence of diffusivity on sodium concentration in the structure.