All Categories → No Category → equivariance

Top 4 equivariance open source projects

geometric-vector-perceptron
Implementation of Geometric Vector Perceptron, a simple circuit for 3d rotation equivariance for learning over large biomolecules, in Pytorch. Idea proposed and accepted at ICLR 2021
egnn-pytorch
Implementation of E(n)-Equivariant Graph Neural Networks, in Pytorch
En-transformer
Implementation of E(n)-Transformer, which extends the ideas of Welling's E(n)-Equivariant Graph Neural Network to attention
1-4 of 4 equivariance projects