lucidrains / slot-attention
Implementation of Slot Attention from GoogleAI
☆410Updated 6 months ago
Alternatives and similar repositories for slot-attention:
Users that are interested in slot-attention are comparing it to the libraries listed below
- Pytorch Implementation of paper "Object-Centric Learning with Slot Attention"☆89Updated last year
- EsViT: Efficient self-supervised Vision Transformers☆411Updated last year
- Repository for "Space-Time Correspondence as a Contrastive Random Walk" (NeurIPS 2020)☆269Updated 3 years ago
- MultiMAE: Multi-modal Multi-task Masked Autoencoders, ECCV 2022☆562Updated 2 years ago
- Masked Siamese Networks for Label-Efficient Learning (https://arxiv.org/abs/2204.07141)☆455Updated 2 years ago
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆257Updated 4 years ago
- PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning☆483Updated 2 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch☆1,120Updated last year
- ☆159Updated 2 years ago
- Code to reproduce the results in the FAIR research papers "Semi-Supervised Learning of Visual Features by Non-Parametrically Predicting V…☆487Updated last year
- A Pytorch-Lightning implementation of self-supervised algorithms☆537Updated 2 years ago
- Multi-object image datasets with ground-truth segmentation masks and generative factors.☆263Updated 3 years ago
- An attempt at the implementation of Glom, Geoffrey Hinton's new idea that integrates concepts from neural fields, top-down-bottom-up proc…☆193Updated 3 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 …☆84Updated 2 years ago
- Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"☆420Updated 2 years ago
- ☆317Updated 11 months ago
- VQVAEs, GumbelSoftmaxes and friends☆553Updated 3 years ago
- NeurIPS 2020, Debiased Contrastive Learning☆282Updated last year
- Understanding Training Dynamics of Deep ReLU Networks☆284Updated 3 weeks ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆137Updated 3 years ago
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,064Updated 2 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆421Updated 4 months ago
- A PyTorch implementation of Perceiver, Perceiver IO and Perceiver AR with PyTorch Lightning scripts for distributed training☆452Updated last year
- ☆245Updated 2 years ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆204Updated last year
- Self-labelling via simultaneous clustering and representation learning. (ICLR 2020)☆539Updated 6 months ago
- Dense Contrastive Learning (DenseCL) for self-supervised representation learning, CVPR 2021 Oral.☆555Updated last year
- VICRegL official code base☆223Updated last year
- Public repo for Augmented Multiscale Deep InfoMax representation learning☆401Updated 4 years ago
- PyTorch code for "Prototypical Contrastive Learning of Unsupervised Representations"☆578Updated 2 years ago