MaximeRobeyns / bayesian_flow_networksLinks
A PyTorch implementation of Bayesian flow networks (Graves et al., 2023).
☆29Updated 2 years ago
Alternatives and similar repositories for bayesian_flow_networks
Users that are interested in bayesian_flow_networks are comparing it to the libraries listed below
Sorting:
- Code for the paper https://arxiv.org/abs/2205.14987v2☆58Updated last year
- This is the official code release for Bayesian Flow Networks.☆309Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆72Updated 10 months ago
- ☆32Updated last year
- Official implementation of the NeurIPS 24 paper of statistical flow matching (SFM) for discrete generation.☆44Updated last year
- Educational implementation of the Discrete Flow Matching paper☆130Updated last year
- Official implementation of Metric Flow Matching (NeurIPS 2024)☆55Updated 5 months ago
- ☆130Updated last year
- Code for the paper https://arxiv.org/abs/2402.04997☆104Updated 2 years ago
- Likelihood Training of Schrödinger Bridge using FBSDEs Theory, ICLR 2022☆88Updated 3 years ago
- PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.☆191Updated 4 years ago
- A demo shows how to combine Langevin dynamics with score matching for generative models.☆42Updated 5 years ago
- Improved diffusion generative models with subspaces☆135Updated 3 years ago
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆84Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆42Updated 2 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆66Updated last year
- Transformers with doubly stochastic attention☆53Updated 3 years ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆70Updated 11 months ago
- A minimalist implementation of score-based diffusion model☆128Updated 4 years ago
- ☆13Updated 2 years ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆26Updated 2 years ago
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆200Updated 11 months ago
- The Superposition of Diffusion Models Using the Itô Density Estimator☆51Updated 10 months ago
- ☆231Updated 2 years ago
- Implementation of Action Matching☆51Updated 2 years ago
- Stochastic Normalizing Flows☆79Updated 4 years ago
- Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)☆153Updated 4 years ago
- Official PyTorch implementation for "Unifying Bayesian Flow Networks and Diffusion Models through Stochastic Differential Equations"☆41Updated last year
- ☆206Updated last year
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆30Updated 11 months ago