Popular GitHub Repositories for Permutation Equivariance

A curated collection of resources and research related to the geometry of representations in the brain, deep networks, and beyond

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

Geometric GNN Dojo provides unified implementations and experiments to explore the design space of Geometric Graph Neural Networks (ICML 2023)

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

A Euclidean diffusion model for structure-based drug design.

Official code of the paper "ReDet: A Rotation-Equivariant Detector for Aerial Object Detection" (CVPR 2021)

[NeurIPS'22] Tokenized Graph Transformer (TokenGT), in PyTorch

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

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

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

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

PiCIE: Unsupervised Semantic Segmentation using Invariance and Equivariance in clustering (CVPR2021)

Adaptation of the merging method described in the paper - Git Re-Basin: Merging Models modulo Permutation Symmetries (https://arxiv.org/abs/2209.04836) for Stable Diffusion

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

Multi-domain Distribution Learning for De Novo Drug Design

SPHARM-Net: Spherical Harmonics-based Convolutional Neural Network

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

Equivariant Subgraph Aggregation Networks (ICLR 2022 Spotlight)

Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).

[CVPR'21] SetVAE: Learning Hierarchical Composition for Generative Modeling of Set-Structured Data, in PyTorch