A method for the prediction of molecular dynamics trajectories using long time steps

We generalize attention and neighborhood attention to the two-dimensional sphere in an equivariant manner.