zhougroup / BAM
Bayesian Attention Modules
☆35Updated 4 years ago
Alternatives and similar repositories for BAM:
Users that are interested in BAM are comparing it to the libraries listed below
- Pytorch implementation of neural processes and variants☆28Updated 8 months ago
- Code accompanying paper: Meta-Learning to Improve Pre-Training☆37Updated 3 years ago
- Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')☆16Updated 2 years ago
- An Empirical Study of Invariant Risk Minimization☆27Updated 4 years ago
- Self-Supervised Learning with Data Augmentations Provably Isolates Content from Style☆51Updated 3 years ago
- Efficient Conditionally Invariant Representation Learning (ICLR 2023, Oral)☆21Updated 2 years ago
- This is reimplementation of "Simple and Principled Uncertainty Estimation with Deterministic Deep Learning via Distance Awareness" in Pyt…☆50Updated 3 years ago
- ☆64Updated 4 years ago
- ☆36Updated 4 years ago
- Code for Knowledge-Adaptation Priors based on the NeurIPS 2021 paper by Khan and Swaroop.☆16Updated 3 years ago
- Training quantile models☆43Updated 4 months ago
- ☆30Updated 3 years ago
- ☆16Updated 2 years ago
- Example code of Sparse Gaussian Process Attention (ICLR 2023)☆24Updated 9 months ago
- [EMNLP'19] Summary for Transformer Understanding☆53Updated 5 years ago
- Noise Contrastive Estimation (NCE) in PyTorch☆32Updated last month
- This is the code for the paper Embrace the Gap: VAEs perform Independent Mechanism Analysis, showing that optimizing the ELBO is equivale…☆23Updated 11 months ago
- Energy-Based Models for Continual Learning Official Repository (PyTorch)☆41Updated 2 years ago
- ☆31Updated 2 years ago
- ☆37Updated 8 months ago
- Featurized Density Ratio Estimation☆20Updated 3 years ago
- Code for "Causal autoregressive flows" - AISTATS, 2021☆44Updated 4 years ago
- Official implementation of Transformer Neural Processes☆74Updated 2 years ago
- Exemplar VAE: Linking Generative Models, Nearest Neighbor Retrieval, and Data Augmentation☆69Updated 4 years ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆40Updated last year
- A study on the following problems: what the memorization problem is in meta-learning; why memorization problem happens; and how we can pr…☆21Updated 2 years ago
- Repository for the NeurIPS 2023 paper "Beyond Confidence: Reliable Models Should Also Consider Atypicality"☆12Updated last year
- VAEs and nonlinear ICA: a unifying framework☆47Updated 5 years ago
- Code for "SAM as an Optimal Relaxation of Bayes", ICLR 2023.☆25Updated last year
- How certain is your transformer?☆25Updated 3 years ago