wzlxjtu / PositionalEncoding2D
A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.
☆250Updated 4 years ago
Alternatives and similar repositories for PositionalEncoding2D:
Users that are interested in PositionalEncoding2D are comparing it to the libraries listed below
- An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow☆561Updated 2 months ago
- Learning Rate Warmup in PyTorch☆399Updated this week
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆418Updated 3 months ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆214Updated 3 years ago
- Implementation of Axial attention - attending to multi-dimensional data efficiently☆361Updated 3 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,115Updated last year
- An All-MLP solution for Vision, from Google AI☆1,007Updated 4 months ago
- An implementation of the efficient attention module.☆297Updated 4 years ago
- Implementation of Slot Attention from GoogleAI☆405Updated 4 months ago
- A pytorch port of google-research/google-research/robust_loss/☆670Updated 3 years ago
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆258Updated 3 years ago
- Fast, differentiable sorting and ranking in PyTorch☆786Updated last year
- Sinkhorn Transformer - Practical implementation of Sparse Sinkhorn Attention☆256Updated 3 years ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence length☆724Updated 8 months ago
- PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning☆480Updated 2 years ago
- [NeurIPS'19] Deep Equilibrium Models☆733Updated 2 years ago
- Geometric loss functions between point clouds, images and volumes☆609Updated 11 months ago
- PyTorch implementation of Barlow Twins.☆966Updated 2 years ago
- Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorc…☆305Updated 3 years ago
- Implementing Stand-Alone Self-Attention in Vision Models using Pytorch☆455Updated 4 years ago
- Official Code for Invertible Residual Networks☆522Updated 5 months ago
- [NeurIPS'20] Multiscale Deep Equilibrium Models☆232Updated 3 years ago
- Pytorch implementation of "An intriguing failing of convolutional neural networks and the CoordConv solution" - https://arxiv.org/abs/180…☆150Updated last year
- Implementation of Linformer for Pytorch☆262Updated last year
- Implementation of the 😇 Attention layer from the paper, Scaling Local Self-Attention For Parameter Efficient Visual Backbones☆198Updated 3 years ago
- Official PyTorch Repo for "ReZero is All You Need: Fast Convergence at Large Depth"☆408Updated 5 months ago
- Approximating Wasserstein distances with PyTorch☆458Updated last year
- Gaussian mixture models in PyTorch.☆535Updated last year
- Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms☆256Updated 3 years ago
- ☆446Updated last year