zhougroup / BAMLinks
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
Sorting:
- Official implementation of Transformer Neural Processes☆78Updated 3 years ago
- Code accompanying paper: Meta-Learning to Improve Pre-Training☆37Updated 3 years ago
- ☆109Updated 3 years ago
- Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')☆16Updated 2 years ago
- ☆16Updated 2 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆66Updated 3 years ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- ☆38Updated last year
- Pytorch implementation of neural processes and variants☆29Updated last year
- Official repository of the paper "Efficient Neural Causal Discovery without Acyclicity Constraints"☆86Updated last year
- Efficient Conditionally Invariant Representation Learning (ICLR 2023, Oral)☆21Updated 2 years ago
- Self-Supervised Learning with Data Augmentations Provably Isolates Content from Style☆53Updated 3 years ago
- This is reimplementation of "Simple and Principled Uncertainty Estimation with Deterministic Deep Learning via Distance Awareness" in Pyt…☆52Updated 4 years ago
- ☆37Updated 4 years ago
- Gradient Estimation with Discrete Stein Operators (NeurIPS 2022)☆17Updated last year
- Code to reproduce the results for Compositional Attention☆59Updated 2 years ago
- Gradient-based Hyperparameter Optimization Over Long Horizons☆14Updated 4 years ago
- ☆25Updated last year
- PyTorch implementation of the NCDSSM models presented in the ICML '23 paper "Neural Continuous-Discrete State Space Models for Irregularl…☆25Updated 2 years ago
- Implementation of the models and datasets used in "An Information-theoretic Approach to Distribution Shifts"☆25Updated 3 years ago
- Deep Learning & Information Bottleneck☆61Updated 2 years ago
- Robust Learning with the Hilbert-Schmidt Independence Criterion☆47Updated 5 years ago
- Low-variance and unbiased gradient for backpropagation through categorical random variables, with application in variational auto-encoder…☆17Updated 5 years ago
- ☆16Updated 3 years ago
- ☆20Updated 5 years ago
- NeurIPS 2022: Tree Mover’s Distance: Bridging Graph Metrics and Stability of Graph Neural Networks☆37Updated 2 years ago
- An Empirical Study of Invariant Risk Minimization☆27Updated 5 years ago
- Repository for Beyond Pinball Loss: Quantile Methods for Calibrated Uncertainty Quantification (NeurIPS 2024)☆43Updated 10 months ago
- Neural Diffusion Processes☆81Updated last year
- ☆21Updated 3 years ago