Spatial attention kinetic network with E(n) equivariance

Yuanqing Wang and John D. Chodera
preprint: [arXiv] [code]

This work descibes Spatial Attention Kinetic Networks (SAKE), a new E(n)-equivariant architecture that uses spatial attention, enabling the construction of extremely performant but still accurate machine learning potentials, as well as flows capable of prediction dynamics.