nnaisense / bayesian-flow-networks
This is the official code release for Bayesian Flow Networks.
☆253Updated 3 months ago
Related projects ⓘ
Alternatives and complementary repositories for bayesian-flow-networks
- A simple implimentation of Bayesian Flow Networks (BFN)☆238Updated 10 months ago
- Annotated Flow Matching paper☆129Updated last month
- A summary of related works about flow matching, stochastic interpolants☆326Updated 3 months ago
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆392Updated 8 months ago
- ☆104Updated 8 months ago
- PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.☆144Updated 2 years ago
- Simplified Masked Diffusion Language Model☆201Updated last month
- Educational implementation of the Discrete Flow Matching paper☆48Updated 2 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆181Updated this week
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆52Updated 2 months ago
- ☆298Updated 2 years ago
- ☆205Updated last year
- Official PyTorch implementation of the paper: Flow Matching in Latent Space☆195Updated last week
- A PyTorch implementation of Bayesian flow networks (Graves et al., 2023).☆22Updated 11 months ago
- code for "Riemannian Flow Matching on General Geometries".☆176Updated 7 months ago
- Implementation of Autoregressive Diffusion in Pytorch☆291Updated this week
- A mini-library for training consistency models.☆213Updated 10 months ago
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆118Updated last year
- A modular, easy to extend GFlowNet library☆231Updated this week
- A curated list for awesome discrete diffusion models resources.☆61Updated this week
- Neural Diffusion Processes☆72Updated 3 months ago
- Score-based generative models for compact manifolds☆103Updated 9 months ago
- TorchCFM: a Conditional Flow Matching library☆1,198Updated last month
- Code for the paper https://arxiv.org/abs/2205.14987v2☆43Updated 6 months ago
- Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)☆134Updated 2 years ago
- Flow-matching algorithms in JAX☆73Updated 2 months ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆122Updated 10 months ago
- Generative Flow Networks - GFlowNet☆182Updated this week
- Modern Fixed Point Systems using Pytorch☆82Updated last year
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆39Updated 7 months ago