Search 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

escnn

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

egnn-pytorch

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

se3-transformer-pytorch

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

equiformer-pytorch

Equiformer - SE3/E3 Graph Attention Transformer for Molecules and Proteins

lie-transformer-pytorch

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

emlp

A library for programmatically generating equivariant layers through constraint solving

geometric-vector-perceptron

Geometric Vector Perceptron - Pytorch

openequivariance-extjax

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

harmonyx

Fully-Differentiable Tensor Spherical Harmonics in JAX

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).

geom3d

Geom3D: Geometric Modeling on 3D Structures, NeurIPS 2023

equiadapt

Library to make any existing neural network architecture equivariant