Update atom gradients during repaint / reset operations

[#47122043]