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☆27Updated 8 months ago
- Code for "SAM as an Optimal Relaxation of Bayes", ICLR 2023.☆25Updated last year
- ☆37Updated 7 months ago
- Official implementation of Transformer Neural Processes☆71Updated 2 years ago
- 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
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆20Updated last year
- This is reimplementation of "Simple and Principled Uncertainty Estimation with Deterministic Deep Learning via Distance Awareness" in Pyt…☆48Updated 3 years ago
- ☆36Updated 4 years ago
- Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')☆16Updated 2 years ago
- Code for "Causal autoregressive flows" - AISTATS, 2021☆44Updated 3 years ago
- Gradient-based Hyperparameter Optimization Over Long Horizons☆13Updated 3 years ago
- PyTorch implementation of the NCDSSM models presented in the ICML '23 paper "Neural Continuous-Discrete State Space Models for Irregularl…☆25Updated last year
- ☆13Updated 2 years ago
- Benchmark for Natural Temporal Distribution Shift (NeurIPS 2022)☆65Updated last year
- ☆20Updated 3 years ago
- Noise Contrastive Estimation (NCE) in PyTorch☆32Updated 3 weeks ago
- ☆108Updated 2 years ago
- An Empirical Study of Invariant Risk Minimization☆27Updated 4 years ago
- Deep Learning & Information Bottleneck☆58Updated last year
- Code repository of the paper "CITRIS: Causal Identifiability from Temporal Intervened Sequences" and "iCITRIS: Causal Representation Lear…☆51Updated last year
- ☆24Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Dirichlet-Variational Auto-Encoder by PyTorch☆22Updated 2 years ago
- VAEs and nonlinear ICA: a unifying framework☆34Updated 4 years ago
- ☆15Updated 2 years ago
- Efficient Conditionally Invariant Representation Learning (ICLR 2023, Oral)☆21Updated 2 years ago
- [EMNLP'19] Summary for Transformer Understanding☆53Updated 5 years ago
- Stochastic Gradient Langevin Dynamics for Bayesian learning☆31Updated 3 years ago
- Official PyTorch implementation of Hyperbolic Neural Networks++☆69Updated 4 years ago