lucidrains / slot-attentionLinks
Implementation of Slot Attention from GoogleAI
☆438Updated 10 months ago
Alternatives and similar repositories for slot-attention
Users that are interested in slot-attention are comparing it to the libraries listed below
Sorting:
- MultiMAE: Multi-modal Multi-task Masked Autoencoders, ECCV 2022☆581Updated 2 years ago
- EsViT: Efficient self-supervised Vision Transformers☆413Updated last year
- Masked Siamese Networks for Label-Efficient Learning (https://arxiv.org/abs/2204.07141)☆455Updated 3 years ago
- Pytorch Implementation of paper "Object-Centric Learning with Slot Attention"☆95Updated last year
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,153Updated last year
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆259Updated 4 years ago
- ☆166Updated 2 years ago
- Repository for "Space-Time Correspondence as a Contrastive Random Walk" (NeurIPS 2020)☆271Updated 3 years ago
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆817Updated 2 years ago
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆477Updated last year
- An attempt at the implementation of Glom, Geoffrey Hinton's new idea that integrates concepts from neural fields, top-down-bottom-up proc…☆194Updated 4 years ago
- This is the official source code for SLATE. We provide the code for the model, the training code, and a dataset loader for the 3D Shapes …☆87Updated 2 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆430Updated 8 months ago
- Dense Contrastive Learning (DenseCL) for self-supervised representation learning, CVPR 2021 Oral.☆560Updated last year
- FiLM: Visual Reasoning with a General Conditioning Layer☆359Updated 3 years ago
- Code to reproduce the results in the FAIR research papers "Semi-Supervised Learning of Visual Features by Non-Parametrically Predicting V…☆488Updated 2 years ago
- [NeurIPS'20] Multiscale Deep Equilibrium Models☆233Updated 3 years ago
- A Pytorch-Lightning implementation of self-supervised algorithms☆539Updated 3 years ago
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,074Updated 2 years ago
- Multi-object image datasets with ground-truth segmentation masks and generative factors.☆269Updated 3 years ago
- Whitening for Self-Supervised Representation Learning | Official repository☆129Updated 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
- PyTorch implementation of Barlow Twins.☆983Updated 3 years ago
- Pytorch implementation of LOST unsupervised object discovery method☆246Updated 2 years ago
- VICRegL official code base☆229Updated 2 years ago
- PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning☆492Updated 3 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆141Updated 3 years ago
- Understanding Training Dynamics of Deep ReLU Networks☆293Updated last month
- PyTorch code for "Prototypical Contrastive Learning of Unsupervised Representations"☆593Updated last month
- Multimodal Masked Autoencoders (M3AE): A JAX/Flax Implementation☆103Updated 3 months ago