tatp22 / multidim-positional-encoding
An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow
☆575Updated 5 months ago
Alternatives and similar repositories for multidim-positional-encoding:
Users that are interested in multidim-positional-encoding are comparing it to the libraries listed below
- A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.☆252Updated 4 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,134Updated last year
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆567Updated 3 months ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆426Updated 5 months ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆751Updated 10 months ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,101Updated 10 months ago
- A collection of literature after or concurrent with Masked Autoencoder (MAE) (Kaiming He el al.).☆814Updated 8 months ago
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆455Updated last year
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆812Updated 2 years ago
- A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)☆718Updated 2 years ago
- An All-MLP solution for Vision, from Google AI☆1,015Updated 6 months ago
- Implementation of Rotary Embeddings, from the Roformer paper, in Pytorch☆651Updated 4 months ago
- An implementation of local windowed attention for language modeling☆429Updated 2 months ago
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,070Updated 2 years ago
- MultiMAE: Multi-modal Multi-task Masked Autoencoders, ECCV 2022☆570Updated 2 years ago
- Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch☆1,799Updated 8 months ago
- An implementation of Performer, a linear attention-based transformer, in Pytorch☆1,119Updated 3 years ago
- ☆450Updated last year
- Implementation of Slot Attention from GoogleAI☆417Updated 7 months ago
- Pytorch library for fast transformer implementations☆1,687Updated 2 years ago
- Explainability for Vision Transformers☆926Updated 3 years ago
- Implementation of Axial attention - attending to multi-dimensional data efficiently☆373Updated 3 years ago
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆781Updated 8 months ago
- PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)☆1,322Updated 9 months ago
- Implementation of Linformer for Pytorch☆274Updated last year
- PyTorch implementation of MoCo v3 https//arxiv.org/abs/2104.02057☆1,256Updated 3 years ago
- PyTorch implementation of SimSiam https//arxiv.org/abs/2011.10566☆1,183Updated 2 years ago
- solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning☆1,472Updated this week
- A simple cross attention that updates both the source and target in one step☆164Updated 10 months ago
- PyTorch implementation of Barlow Twins.☆975Updated 3 years ago