We calculated MSD and their uncertainties over a time scale of 15.4 ps, averaging over each set of 10 independent runs obtained with AIMD and MLMD. We excluded the first 300 steps of each simulation to ensure that only diffusive motion was investigated. The MSD benchmark shows quantitatively similar behaviour for both AIMD and MLMD simulations, indicating that diffusion is well described at high temperatures.