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

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

escnn

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

equiformer-pytorch

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

openequivariance-extjax

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

emlp

A library for programmatically generating equivariant layers through constraint solving

lie-transformer-pytorch

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

equiadapt

Library to make any existing neural network architecture equivariant

tmeasures

The Transformational Measures (TM) library allows neural network researchers to evaluate the invariance and equivariance of their models with respect to a set of transformations. Support for Pytorch (current) and Tensorflow/Keras (coming).

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

geom3d

Geom3D: Geometric Modeling on 3D Structures, NeurIPS 2023

harmonyx

Fully-Differentiable Tensor Spherical Harmonics in JAX