lucidrains / lie-transformer-pytorchLinks
Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch
☆96Updated 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
Sorting:
- Implementation of Geometric Vector Perceptron, a simple circuit for 3d rotation equivariance for learning over large biomolecules, in Pyt…☆76Updated 4 years ago
- Equivariant Transformer (ET) layers are image-to-image mappings that incorporate prior knowledge on invariances with respect to continuo…☆93Updated 6 years ago
- ☆70Updated 2 years ago
- Graph neural network message passing reframed as a Transformer with local attention☆70Updated 2 years ago
- Authors implementation of LieTransformer: Equivariant Self-Attention for Lie Groups☆36Updated 4 years ago
- Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network☆225Updated last year
- LieTransformer: Equivariant Self-Attention for Lie Groups☆74Updated last year
- An implementation of (Induced) Set Attention Block, from the Set Transformers paper☆65Updated 2 years ago
- Implementation of Tranception, an attention network, paired with retrieval, that is SOTA for protein fitness prediction☆32Updated 3 years ago
- Energy-based models for atomic-resolution protein conformations☆100Updated 3 years ago
- To be a next-generation DL-based phenotype prediction from genome mutations.☆19Updated 4 years ago
- Deterministic Decoding for Discrete Data in Variational Autoencoders☆24Updated 5 years ago
- Lie Algebra Convolutional Network implementation☆43Updated 3 years ago
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆36Updated 4 years ago
- Implementation of deep implicit attention in PyTorch☆65Updated 4 years ago
- ☆39Updated 3 years ago
- Code for "Bridging the Gap between f-GANs and Wasserstein GANs", ICML 2020☆14Updated 5 years ago
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆59Updated 4 years ago
- This repository implements and evaluates convolutional networks on the Möbius strip as toy model instantiations of Coordinate Independent…☆74Updated last year
- Pytorch reimplementation of Molecule Attention Transformer, which uses a transformer to tackle the graph-like structure of molecules☆58Updated 4 years ago
- An implementation of 2021 paper by Geoffrey Hinton: "How to represent part-whole hierarchies in a neural network" in Pytorch.☆57Updated 4 years ago
- Implementation of Kronecker Attention in Pytorch☆19Updated 5 years ago
- Code for UAI 2020 paper "Locally Masked Convolution for Autoregressive Models"☆79Updated 5 years ago
- Sample pytorch implementation of Covariant Compositional Networks☆13Updated 7 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
- A simple Transformer where the softmax has been replaced with normalization☆20Updated 5 years ago
- Exemplar VAE: Linking Generative Models, Nearest Neighbor Retrieval, and Data Augmentation☆69Updated 4 years ago
- ☆42Updated 2 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆66Updated 3 years ago
- ☆30Updated 5 years ago