apple / ml-probabilistic-attentionLinks
☆33Updated 3 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
Sorting:
- TLDR is an unsupervised dimensionality reduction method that combines neighborhood embedding learning with the simplicity and effectivene…☆127Updated 3 years ago
- Meta Optimal Transport☆107Updated 2 years ago
- Official implementation of the paper "Topographic VAEs learn Equivariant Capsules"☆81Updated 3 years ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆51Updated 3 years ago
- Official repository for the paper "Masksembles for Uncertainty Estimation" (CVPR 2021).☆103Updated 3 months ago
- ☆88Updated 2 years ago
- Visual Representation Learning Benchmark for Self-Supervised Models☆35Updated last year
- PyTorch implementation of IRMAE https//arxiv.org/abs/2010.00679☆48Updated 3 years ago
- (ECCV 2022) BayesCap: Bayesian Identity Cap for Calibrated Uncertainty in Frozen Neural Networks☆51Updated 3 years ago
- Official code for `Visual Attention Emerges from Recurrent Sparse Reconstruction' (ICML 2022)☆36Updated 3 years ago
- Implementation of Kronecker Attention in Pytorch☆19Updated 5 years ago
- An implementation of (Induced) Set Attention Block, from the Set Transformers paper☆67Updated 3 years ago
- Github code for the paper Maximum Class Separation as Inductive Bias in One Matrix. Arxiv link: https://arxiv.org/abs/2206.08704☆30Updated 2 years ago
- [CogSci'21] Study of human inductive biases in CNNs and Transformers.☆43Updated 4 years ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆116Updated 3 years ago
- ICLR22 "Fast Differentiable Matrix Square Root" and T-PAMI extension☆67Updated last year
- Official implementation of "Energy-Based Models for Deep Probabilistic Regression" (ECCV 2020) and "How to Train Your Energy-Based Model …☆106Updated 5 years ago
- ☆39Updated last year
- An implementation of 2021 paper by Geoffrey Hinton: "How to represent part-whole hierarchies in a neural network" in Pytorch.☆57Updated 4 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆57Updated 3 years ago
- Code for the paper Self-Supervised Learning of Split Invariant Equivariant Representations☆31Updated 2 years ago
- Implementation of OmniNet, Omnidirectional Representations from Transformers, in Pytorch☆59Updated 4 years ago
- A project to improve out-of-distribution detection (open set recognition) and uncertainty estimation by changing a few lines of code in y…☆44Updated 3 years ago
- ☆36Updated 2 years ago
- ☆46Updated last year
- A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.☆49Updated 5 months ago
- Differentiable Top-k Classification Learning☆92Updated 3 years ago
- ☆47Updated 3 years ago
- Stochastic Optimization for Global Contrastive Learning without Large Mini-batches☆20Updated 2 years ago
- Official code for NeurIPS paper "Combinatorial Optimization for Panoptic Segmentation: A Fully Differentiable Approach".☆16Updated 3 years ago