lucidrains / lie-transformer-pytorch
Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch
☆90Updated 4 years ago
Alternatives and similar repositories for lie-transformer-pytorch:
Users that are interested in lie-transformer-pytorch are comparing it to the libraries listed below
- Implementation of Geometric Vector Perceptron, a simple circuit for 3d rotation equivariance for learning over large biomolecules, in Pyt…☆73Updated 3 years ago
- Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network☆219Updated 10 months ago
- LieTransformer: Equivariant Self-Attention for Lie Groups☆69Updated 8 months ago
- Equivariant Transformer (ET) layers are image-to-image mappings that incorporate prior knowledge on invariances with respect to continuo…☆90Updated 5 years ago
- Lie Algebra Convolutional Network implementation☆43Updated 3 years ago
- Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups☆36Updated 4 years ago
- Code repository for the paper "Group Equivariant Stand-Alone Self Attention For Vision" published at ICLR 2021. https://openreview.net/fo…☆29Updated 4 years ago
- Energy-based models for atomic-resolution protein conformations☆97Updated 3 years ago
- This repository implements and evaluates convolutional networks on the Möbius strip as toy model instantiations of Coordinate Independent…☆72Updated last year
- Implementation of Invariant Point Attention, used for coordinate refinement in the structure module of Alphafold2, as a standalone Pytorc…☆162Updated 2 years ago
- ☆69Updated 2 years ago
- [ICLR 2020] FSPool: Learning Set Representations with Featurewise Sort Pooling☆42Updated last year
- Graph neural network message passing reframed as a Transformer with local attention☆68Updated 2 years ago
- ☆157Updated 5 years ago
- ICML 2020 Paper: Latent Variable Modelling with Hyperbolic Normalizing Flows☆53Updated 2 years ago
- Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch. This specific repository is geared towards integration wit…☆281Updated 2 years ago
- A Prior of a Googol Gaussians: a Tensor Ring Induced Prior for Generative Models☆27Updated 9 months ago
- ☆37Updated 2 years ago
- Pytorch reimplementation of Molecule Attention Transformer, which uses a transformer to tackle the graph-like structure of molecules☆58Updated 4 years ago
- Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"☆22Updated 2 years ago
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆36Updated 4 years ago
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆55Updated 3 years ago
- TensorFlow implementation of "noisy K-FAC" and "noisy EK-FAC".☆60Updated 6 years ago
- A TensorFlow implementation of the paper 'Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks'☆31Updated 11 months ago
- GraphNVP: An Invertible Flow Model for Generating Molecular Graphs☆92Updated 2 years ago
- pytorch implementation of trDesign☆45Updated 4 years ago
- Implementation of deep implicit attention in PyTorch☆65Updated 3 years ago
- Code for A General Recipe for Likelihood-free Bayesian Optimization, ICML 2022☆44Updated 2 years ago
- Tensorflow implementation and notebooks for Implicit Maximum Likelihood Estimation☆67Updated 3 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆44Updated 2 years ago