equivariance

Implementation of DiffDock: Diffusion Steps, Twists, and Turns for Molecular Docking

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

E(2)-Equivariant CNNs Library for Pytorch

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

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

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

A Euclidean diffusion model for structure-based drug design.

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

DiffLinker: Equivariant 3D-Conditional Diffusion Model for Molecular Linker Design

[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 Torsional Diffusion for Molecular Conformer Generation (NeurIPS 2022)

A library for programmatically generating equivariant layers through constraint solving

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

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

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

Multi-domain Distribution Learning for De Novo Drug Design

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

[ICCV 2025 Highlight] ETCH: Generalizing Body Fitting to Clothed Humans via Equivariant Tightness