Equivariance Python Packages

openequivariance

OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.

e2cnn

E(2)-Equivariant CNNs Library for Pytorch

escnn

Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/

en-transformer

Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network

se3-transformer-pytorch

Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch. This specific repository is geared towards integration with eventual Alphafold2 replication.

egnn-pytorch

Implementation of E(n)-Equivariant Graph Neural Networks, in Pytorch

lie-transformer-pytorch

Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch

equiformer-pytorch

Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein folding

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

openequivariance-extjax

OpenEquivariance: a fast, open-source GPU JIT kernel generator for the Clebsch-Gordon Tensor Product.

emlp

A Practical Method for Constructing Equivariant Multilayer Perceptrons for Arbitrary Matrix Groups

geom3d

Geom3D: Geometric Modeling on 3D Structures, NeurIPS 2023

tmeasures

Invariance, Same-Equivariance and other measures for Neural Networks. Support for PyTorch (now) and TensorFlow (coming).

equiadapt

Library to make any existing neural network architecture equivariant

harmonyx

Fully-differentiable tensor spherical harmonics in JAX