lucidrains / adjacent-attention-networkView external linksLinks
Graph neural network message passing reframed as a Transformer with local attention
☆70Dec 24, 2022Updated 3 years ago
Alternatives and similar repositories for adjacent-attention-network
Users that are interested in adjacent-attention-network are comparing it to the libraries listed below
Sorting:
- An implementation of (Induced) Set Attention Block, from the Set Transformers paper☆67Jan 10, 2023Updated 3 years ago
- Pytorch reimplementation of Molecule Attention Transformer, which uses a transformer to tackle the graph-like structure of molecules☆58Dec 2, 2020Updated 5 years ago
- Implementation of Cross Transformer for spatially-aware few-shot transfer, in Pytorch☆54Mar 30, 2021Updated 4 years ago
- Implementation of E(n)-Transformer, which incorporates attention mechanisms into Welling's E(n)-Equivariant Graph Neural Network☆226Jun 2, 2024Updated last year
- Implementation of the Triangle Multiplicative module, used in Alphafold2 as an efficient way to mix rows or columns of a 2d feature map, …☆39Aug 3, 2021Updated 4 years ago
- ☆12Mar 16, 2022Updated 3 years ago
- Implementation of the Remixer Block from the Remixer paper, in Pytorch☆36Sep 27, 2021Updated 4 years ago
- Implementation of Invariant Point Attention, used for coordinate refinement in the structure module of Alphafold2, as a standalone Pytorc…☆171Nov 25, 2022Updated 3 years ago
- A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network☆150Mar 12, 2021Updated 4 years ago
- Implementation of COCO-LM, Correcting and Contrasting Text Sequences for Language Model Pretraining, in Pytorch☆46Mar 3, 2021Updated 4 years ago
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆49Jan 27, 2022Updated 4 years ago
- Implementation of the retriever distillation procedure as outlined in the paper "Distilling Knowledge from Reader to Retriever"☆32Dec 16, 2020Updated 5 years ago
- Pytorch implementation of Compressive Transformers, from Deepmind☆163Oct 4, 2021Updated 4 years ago
- Local Attention - Flax module for Jax☆22May 26, 2021Updated 4 years ago
- Implementation of OmniNet, Omnidirectional Representations from Transformers, in Pytorch☆59Mar 19, 2021Updated 4 years ago
- Implementation of Metaformer, but in an autoregressive manner☆26Jun 21, 2022Updated 3 years ago
- Probabilistic inference for models of behaviour☆10Oct 13, 2025Updated 4 months ago
- Implementation of the DDPM + IPA (invariant point attention) for protein generation, as outlined in the paper "Protein Structure and Sequ…☆89Jul 25, 2022Updated 3 years ago
- machine translation data process tools☆10Apr 29, 2024Updated last year
- ☆11Nov 27, 2022Updated 3 years ago
- A simple Python project that takes a CT scan in DICOM format and outputs a directional X-ray projection.☆11Jun 13, 2019Updated 6 years ago
- ☆13Apr 16, 2022Updated 3 years ago
- Implementation of Insertion-deletion Denoising Diffusion Probabilistic Models☆30May 31, 2022Updated 3 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆99Jan 13, 2021Updated 5 years ago
- ☆10Sep 13, 2021Updated 4 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆57Jan 5, 2023Updated 3 years ago
- Easy Multiprocessing for Python☆42Nov 3, 2020Updated 5 years ago
- ☆13Feb 13, 2023Updated 3 years ago
- Interactively evolve various types of art (pictures, animations, shapes, and sounds) using Compositional Pattern Producing Networks☆21May 9, 2018Updated 7 years ago
- Implementation of Feedback Transformer in Pytorch☆108Mar 2, 2021Updated 4 years ago
- Protein quality assessment using Graph Convolutional Networks☆28Oct 11, 2022Updated 3 years ago
- Implementation of Nvidia's NeuralPlexer, for end-to-end differentiable design of functional small-molecules and ligand-binding proteins, …☆52Nov 20, 2023Updated 2 years ago
- ☆13Nov 12, 2018Updated 7 years ago
- Implementation of π-GAN, for 3d-aware image synthesis, in Pytorch☆124Feb 22, 2021Updated 4 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆220Feb 13, 2023Updated 3 years ago
- Implementation of ProteinBERT in Pytorch☆164Aug 10, 2021Updated 4 years ago
- Implementation of the 😇 Attention layer from the paper, Scaling Local Self-Attention For Parameter Efficient Visual Backbones☆201Mar 24, 2021Updated 4 years ago
- Implementation of Denoising Diffusion for protein design, but using the new Equiformer (successor to SE3 Transformers) with some addition…☆57Dec 27, 2022Updated 3 years ago
- A repository with exploration into using transformers to predict DNA ↔ transcription factor binding☆88Jun 2, 2022Updated 3 years ago