lucidrains / lie-transformer-pytorch
Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch
☆87Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for lie-transformer-pytorch
- Implementation of Geometric Vector Perceptron, a simple circuit for 3d rotation equivariance for learning over large biomolecules, in Pyt…☆72Updated 3 years ago
- Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network☆211Updated 5 months ago
- Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups☆37Updated 3 years ago
- LieTransformer: Equivariant Self-Attention for Lie Groups☆63Updated 3 months ago
- Equivariant Transformer (ET) layers are image-to-image mappings that incorporate prior knowledge on invariances with respect to continuo…☆88Updated 5 years ago
- Code repository for the paper "Group Equivariant Stand-Alone Self Attention For Vision" published at ICLR 2021. https://openreview.net/fo…☆28Updated 3 years ago
- Energy-based models for atomic-resolution protein conformations☆95Updated 2 years ago
- Lie Algebra Convolutional Network implementation☆42Updated 2 years ago
- Deterministic Decoding for Discrete Data in Variational Autoencoders☆24Updated 4 years ago
- Implementation of Invariant Point Attention, used for coordinate refinement in the structure module of Alphafold2, as a standalone Pytorc…☆148Updated last year
- A Prior of a Googol Gaussians: a Tensor Ring Induced Prior for Generative Models☆27Updated 4 months ago
- pytorch implementation of trDesign☆45Updated 3 years ago
- Implementation of Tranception, an attention network, paired with retrieval, that is SOTA for protein fitness prediction☆31Updated 2 years ago
- To be a next-generation DL-based phenotype prediction from genome mutations.☆19Updated 3 years ago
- An implementation of (Induced) Set Attention Block, from the Set Transformers paper☆55Updated last year
- A TensorFlow implementation of the paper 'Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks'☆30Updated 6 months ago
- Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch. This specific repository is geared towards integration wit…☆259Updated last year
- ICML 2020 Paper: Latent Variable Modelling with Hyperbolic Normalizing Flows☆54Updated last year
- This repository implements and evaluates convolutional networks on the Möbius strip as toy model instantiations of Coordinate Independent…☆73Updated 11 months ago
- ☆32Updated 3 years ago
- Graph neural network message passing reframed as a Transformer with local attention☆66Updated last year
- Pytorch reimplementation of Molecule Attention Transformer, which uses a transformer to tackle the graph-like structure of molecules☆58Updated 3 years ago
- Official implementation of the paper "Topographic VAEs learn Equivariant Capsules"☆77Updated 2 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆62Updated 2 years ago
- ☆49Updated 4 years ago
- ☆35Updated last year
- ☆32Updated 2 years ago
- Official repository for "Categorical Normalizing Flows via Continuous Transformations"☆55Updated 3 years ago