heiidii / en-transformer_multiatom Goto Github PK
View Code? Open in Web Editor NEWThis project forked from lucidrains/en-transformer
Implementation of E(n)-Transformer, which extends the ideas of Welling's E(n)-Equivariant Graph Neural Network to attention
License: MIT License