blazingsiyan / En-transformer

Implementation of E(n)-Transformer, which extends the ideas of Welling's E(n)-Equivariant Graph Neural Network to attention

Geek Repo

Github PK Tool