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
- Example code of Sparse Gaussian Process Attention (ICLR 2023)☆23Updated 7 months ago
- Pytorch implementation of neural processes and variants☆27Updated 6 months ago
- ☆16Updated 2 years ago
- Official implementation of Transformer Neural Processes☆71Updated 2 years ago
- Code for Accelerated Linearized Laplace Approximation for Bayesian Deep Learning (ELLA, NeurIPS 22')☆16Updated 2 years ago
- ☆15Updated last year
- Official code for Generative Marginalization Models [ICML 2024] [SPGIM 2023 Workshop Oral]☆22Updated 6 months ago
- Noise Contrastive Estimation (NCE) in PyTorch☆32Updated last year
- Neural Diffusion Processes☆76Updated 6 months ago
- Stochastic Gradient Langevin Dynamics for Bayesian learning☆30Updated 3 years ago
- PyTorch implementation of the NCDSSM models presented in the ICML '23 paper "Neural Continuous-Discrete State Space Models for Irregularl…☆24Updated last year
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆20Updated 11 months ago
- Code for "Causal autoregressive flows" - AISTATS, 2021☆44Updated 3 years ago
- Code repository of the paper "CITRIS: Causal Identifiability from Temporal Intervened Sequences" and "iCITRIS: Causal Representation Lear…☆50Updated 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
- ☆37Updated 6 months ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation (NeurIPS'22).☆20Updated 5 months ago
- VAEs and nonlinear ICA: a unifying framework☆33Updated 4 years ago
- NeurIPS'23: Energy Discrepancies: A Score-Independent Loss for Energy-Based Models☆14Updated 3 months ago
- How certain is your transformer?☆25Updated 3 years ago
- Featurized Density Ratio Estimation☆20Updated 3 years ago
- PyTorch implementation of the mixture distribution family with implicit reparametrisation gradients.☆19Updated last year
- ☆22Updated last year
- Experiments to reproduce results in Interventional Causal Representation Learning.☆25Updated 2 years ago
- Code for "SAM as an Optimal Relaxation of Bayes", ICLR 2023.☆25Updated last year
- Official code for the ICLR 2021 paper Neural ODE Processes☆70Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year
- Code for the paper "Disentanglement by Nonlinear ICA with General Incompressible-flow Networks (GIN)" (2020)☆32Updated 3 years ago
- This is the official source code for Sequential Neural Processes.☆40Updated 2 years ago
- Low-variance and unbiased gradient for backpropagation through categorical random variables, with application in variational auto-encoder…☆17Updated 4 years ago