apple / ml-probabilistic-attention
☆31Updated 2 years ago
Alternatives and similar repositories for ml-probabilistic-attention:
Users that are interested in ml-probabilistic-attention are comparing it to the libraries listed below
- ☆11Updated 2 years ago
- ☆46Updated 8 months ago
- An implementation of (Induced) Set Attention Block, from the Set Transformers paper☆56Updated 2 years ago
- DUET: 2D Structured and Approximately Equivariant Representations, ICML 2023☆18Updated last year
- Source code for the Nature Machine Intelligence paper: When and how convolutional neural networks generalize to out-of-distribution categ…☆22Updated 3 years ago
- ☆37Updated 7 months ago
- We investigated corruption robustness across different architectures including Convolutional Neural Networks, Vision Transformers, and th…☆16Updated 3 years ago
- Research publication code for "Forward Compatible Training for Large-Scale Embedding Retrieval Systems", CVPR 2022, and "FastFill: Effici…☆55Updated 2 years ago
- Directed masked autoencoders☆14Updated 2 years ago
- ☆10Updated last year
- ☆29Updated 3 years ago
- ☆28Updated last year
- TLDR is an unsupervised dimensionality reduction method that combines neighborhood embedding learning with the simplicity and effectivene…☆124Updated 2 years ago
- A Benchmark for Efficient and Compositional Visual Reasoning☆24Updated last year
- Github code for the paper Maximum Class Separation as Inductive Bias in One Matrix. Arxiv link: https://arxiv.org/abs/2206.08704☆29Updated last year
- Making Heads or Tails Towards Semantically Consistent Visual Counterfactuals☆30Updated 2 years ago
- A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.☆49Updated 9 months ago
- ☆21Updated last year
- Official code for NeurIPS paper "Combinatorial Optimization for Panoptic Segmentation: A Fully Differentiable Approach".☆16Updated 2 years ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆44Updated 2 years ago
- Improving Transformation Invariance in Contrastive Representation Learning☆13Updated 4 years ago
- Official implementation of the paper "Provable Stochastic Optimization for Global Contrastive Learning: Small Batch Does Not Harm Perform…☆20Updated last year
- ☆51Updated 9 months ago
- Un-*** 50 billions multimodality dataset☆24Updated 2 years ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆50Updated 2 years ago
- Implementation of Kronecker Attention in Pytorch☆18Updated 4 years ago
- Official implementation of the paper "Topographic VAEs learn Equivariant Capsules"☆79Updated 3 years ago
- ☆76Updated 2 years ago
- An adaptive training algorithm for residual network☆15Updated 4 years ago
- PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations☆16Updated 4 years ago