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 7 months ago
- Official implementation of Transformer Neural Processes☆71Updated 2 years ago
- ☆37Updated 7 months ago
- Code repository of the paper "CITRIS: Causal Identifiability from Temporal Intervened Sequences" and "iCITRIS: Causal Representation Lear…☆50Updated last year
- Experiments to reproduce results in Interventional Causal Representation Learning.☆26Updated 2 years ago
- ☆21Updated 2 years ago
- Example code of Sparse Gaussian Process Attention (ICLR 2023)☆24Updated 8 months ago
- Neural Diffusion Processes☆76Updated 7 months ago
- PyTorch implementation of the NCDSSM models presented in the ICML '23 paper "Neural Continuous-Discrete State Space Models for Irregularl…☆24Updated last year
- Code accompanying paper: Meta-Learning to Improve Pre-Training☆37Updated 3 years ago
- ☆30Updated 3 years ago
- PyTorch implementation of the mixture distribution family with implicit reparametrisation gradients.☆19Updated last year
- 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
- ☆36Updated 4 years ago
- ☆16Updated 2 years ago
- Training quantile models☆43Updated 3 months ago
- Integrated gradients attribution method implemented in PyTorch☆26Updated 4 years ago
- ☆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
- Noise Contrastive Estimation (NCE) in PyTorch☆32Updated last week
- Efficient Conditionally Invariant Representation Learning (ICLR 2023, Oral)☆21Updated 2 years ago
- ☆15Updated 2 years ago
- ☆20Updated 4 years ago
- ☆20Updated 3 years ago
- ☆35Updated 4 years ago
- Stochastic Gradient Langevin Dynamics for Bayesian learning☆30Updated 3 years ago
- Gradient Estimation with Discrete Stein Operators (NeurIPS 2022)☆17Updated last year
- Benchmark for Natural Temporal Distribution Shift (NeurIPS 2022)☆65Updated last year
- Low-variance and unbiased gradient for backpropagation through categorical random variables, with application in variational auto-encoder…☆17Updated 4 years ago