apple / ml-probabilistic-attentionLinks
☆32Updated 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:
- ☆46Updated last year
- TLDR is an unsupervised dimensionality reduction method that combines neighborhood embedding learning with the simplicity and effectivene…☆126Updated 3 years ago
- Official implementation of the paper "Topographic VAEs learn Equivariant Capsules"☆81Updated 3 years ago
- Meta Optimal Transport☆105Updated 2 years ago
- An implementation of (Induced) Set Attention Block, from the Set Transformers paper☆65Updated 2 years ago
- (ECCV 2022) BayesCap: Bayesian Identity Cap for Calibrated Uncertainty in Frozen Neural Networks☆51Updated 3 years ago
- ☆38Updated last year
- Official repository for the paper "Masksembles for Uncertainty Estimation" (CVPR 2021).☆103Updated last month
- Visual Representation Learning Benchmark for Self-Supervised Models☆35Updated last year
- An adaptive training algorithm for residual network☆17Updated 5 years ago
- NeurIPS 2021, Code for Measuring Generalization with Optimal Transport☆28Updated 4 years ago
- ICLR22 "Fast Differentiable Matrix Square Root" and T-PAMI extension☆64Updated last year
- [CogSci'21] Study of human inductive biases in CNNs and Transformers.☆43Updated 4 years ago
- An implementation of 2021 paper by Geoffrey Hinton: "How to represent part-whole hierarchies in a neural network" in Pytorch.☆57Updated 4 years ago
- Differentiable Optimizers with Perturbations in Pytorch☆69Updated 4 years ago
- Official Code: Estimating Model Uncertainty of Neural Networks in Sparse Information Form, ICML2020.☆31Updated 4 years ago
- An official PyTorch implementation of "Regression Prior Networks" for effective runtime uncertainty estimation.☆36Updated 5 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
- Implementation of Kronecker Attention in Pytorch☆19Updated 5 years ago
- Source code for the Nature Machine Intelligence paper: When and how convolutional neural networks generalize to out-of-distribution categ…☆23Updated 3 years ago
- Differentiable Top-k Classification Learning☆91Updated 2 years ago
- Code for the paper Self-Supervised Learning of Split Invariant Equivariant Representations☆30Updated 2 years ago
- Code for "SAM as an Optimal Relaxation of Bayes", ICLR 2023.☆25Updated 2 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
- Uncertainty-aware representation learning (URL) benchmark☆105Updated 9 months ago
- [ICML 2020] Differentiating through the Fréchet Mean (https://arxiv.org/abs/2003.00335).☆59Updated 4 years ago
- ☆42Updated 2 years ago
- [ICML2023] InfoOT: Information Maximizing Optimal Transport☆42Updated 2 years ago
- Code for ICML2021 paper 'Commutative Lie Group VAE for Disentanglement Learning'.☆23Updated 3 years ago
- Making Heads or Tails Towards Semantically Consistent Visual Counterfactuals☆30Updated 3 years ago