Algomancer / Bayesian-Flow-Networks
A simple implimentation of Bayesian Flow Networks (BFN)
☆240Updated last year
Alternatives and similar repositories for Bayesian-Flow-Networks:
Users that are interested in Bayesian-Flow-Networks are comparing it to the libraries listed below
- This is the official code release for Bayesian Flow Networks.☆272Updated 8 months ago
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆530Updated last year
- A mini-library for training consistency models.☆238Updated last year
- A curated list for awesome discrete diffusion models resources.☆268Updated last month
- Annotated Flow Matching paper☆175Updated 6 months ago
- Minimal Implementation of a D3PM in pytorch☆205Updated 11 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated last year
- ☆310Updated 2 years ago
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆201Updated last year
- Simple and Effective Masked Diffusion Language Model☆346Updated 2 weeks ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆67Updated 3 weeks ago
- Educational implementation of the Discrete Flow Matching paper☆79Updated 6 months ago
- code for "Riemannian Flow Matching on General Geometries".☆206Updated last year
- Code for ICML 2023 paper, "PFGM++: Unlocking the Potential of Physics-Inspired Generative Models"☆370Updated last year
- Visualizations of the theory behind diffusion models.☆148Updated 11 months ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆128Updated last year
- ☆301Updated 9 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆112Updated last month
- ☆149Updated 3 months ago
- Experiments around a simple idea for inducing multiple hierarchical predictive model within a GPT☆210Updated 7 months ago
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆263Updated last month
- Unofficial Implementation of Consistency Models in pytorch☆254Updated 2 years ago
- Flow-matching algorithms in JAX☆86Updated 7 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆361Updated last month
- Use Jax functions in Pytorch☆238Updated last year
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆120Updated 7 months ago
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆715Updated 10 months ago
- Score-Based Generative Modeling with Critically-Damped Langevin Diffusion☆197Updated last year
- Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"☆338Updated 2 years ago
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆147Updated 2 years ago