apple / ml-probabilistic-attentionLinks
☆31Updated 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:
- ☆11Updated 3 years ago
- ☆46Updated 11 months ago
- An implementation of (Induced) Set Attention Block, from the Set Transformers paper☆59Updated 2 years ago
- A JAX implementation of Broaden Your Views for Self-Supervised Video Learning, or BraVe for short.☆48Updated last year
- ☆29Updated last year
- ☆76Updated 3 years ago
- Code for ICML2021 paper 'Commutative Lie Group VAE for Disentanglement Learning'.☆22Updated 2 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
- Research publication code for "Forward Compatible Training for Large-Scale Embedding Retrieval Systems", CVPR 2022, and "FastFill: Effici…☆55Updated 2 years ago
- DUET: 2D Structured and Approximately Equivariant Representations, ICML 2023☆18Updated last year
- Making Heads or Tails Towards Semantically Consistent Visual Counterfactuals☆30Updated 2 years ago
- Implementation of Kronecker Attention in Pytorch☆19Updated 4 years ago
- Implementation of Metaformer, but in an autoregressive manner☆25Updated 3 years ago
- An implementation of 2021 paper by Geoffrey Hinton: "How to represent part-whole hierarchies in a neural network" in Pytorch.☆58Updated 4 years ago
- (ECCV 2022) BayesCap: Bayesian Identity Cap for Calibrated Uncertainty in Frozen Neural Networks☆49Updated 2 years ago
- ☆47Updated 2 years ago
- An official PyTorch implementation of "Regression Prior Networks" for effective runtime uncertainty estimation.☆36Updated 4 years ago
- Github for the conference paper GLOD-Gaussian Likelihood OOD detector☆16Updated 3 years ago
- Repo reproducing experimental results in "Addressing the Topological Defects of Disentanglement"☆22Updated 2 years ago
- Implementation of OmniNet, Omnidirectional Representations from Transformers, in Pytorch☆58Updated 4 years ago
- ☆41Updated 2 years ago
- ☆27Updated last month
- Towards Nonlinear Disentanglement in Natural Data with Temporal Sparse Coding☆73Updated 3 years ago
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Updated 2 years ago
- ☆30Updated 3 years ago
- We investigated corruption robustness across different architectures including Convolutional Neural Networks, Vision Transformers, and th…☆16Updated 3 years ago
- A Benchmark for Efficient and Compositional Visual Reasoning☆25Updated last year
- Usable implementation of Emerging Symbol Binding Network (ESBN), in Pytorch☆25Updated 4 years ago
- Official PyTorch implementation of A Quaternion-Valued Variational Autoencoder (QVAE).☆31Updated 3 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆54Updated 2 years ago