equivariance topic

List equivariance repositories

DEVIANT

196
Stars
27
Forks
Watchers

[ECCV 2022] Official PyTorch Code of DEVIANT: Depth Equivariant Network for Monocular 3D Object Detection

EquiBind

464
Stars
110
Forks
Watchers

EquiBind: geometric deep learning for fast predictions of the 3D structure in which a small molecule binds to a protein

equidock_public

223
Stars
59
Forks
Watchers

EquiDock: geometric deep learning for fast rigid 3D protein-protein docking

e2cnn

584
Stars
75
Forks
Watchers

E(2)-Equivariant CNNs Library for Pytorch

egnn-pytorch

387
Stars
65
Forks
Watchers

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

se3-transformer-pytorch

243
Stars
23
Forks
Watchers

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

equivariant-MLP

249
Stars
19
Forks
Watchers

A library for programmatically generating equivariant layers through constraint solving

En-transformer

205
Stars
28
Forks
Watchers

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

simple-equivariant-gnn

120
Stars
12
Forks
Watchers

A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks

wavelet_networks

79
Stars
5
Forks
Watchers

Code repository of the paper "Wavelet Networks: Scale-Translation Equivariant Learning From Raw Time-Series, TMLR" https://arxiv.org/abs/2006.05259