equivariance topic

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

lucidrains

Equivariant Transformer (ET) layers are image-to-image mappings that incorporate prior knowledge on invariances with respect to continuous transformations groups (ICML 2019). Paper: https://arxiv.org...

stanford-futuredata

Implementation of Torsional Diffusion for Molecular Conformer Generation (NeurIPS 2022)

gcorso

Equivariant Subgraph Aggregation Networks (ICLR 2022 Spotlight)

beabevi

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

lucidrains

[CVPR 2022] Official PyTorch implementation of "Official Self-Supervised Equivariant Learning for Oriented Keypoint Detection"

bluedream1121

Official source code for "Roto-translated Local Coordinate Frames For Interacting Dynamical Systems". In NeurIPS 2021.

mkofinas

Code repository for the paper "Attentive Group Equivariant Convolutional Neural Networks" published at ICML 2020. https://arxiv.org/abs/2002.03830

dwromero

A Euclidean diffusion model for structure-based drug design.

arneschneuing

Understanding and Extending Subgraph GNNs by Rethinking their Symmetries (NeurIPS 2022 Oral)

beabevi