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 9 months ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆78Updated 2 months ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆128Updated last year
- A curated list for awesome discrete diffusion models resources.☆309Updated 3 months ago
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆565Updated last year
- A mini-library for training consistency models.☆242Updated last year
- Educational implementation of the Discrete Flow Matching paper☆87Updated 8 months ago
- WIP☆93Updated 8 months ago
- Annotated Flow Matching paper☆182Updated 7 months ago
- Experiments around a simple idea for inducing multiple hierarchical predictive model within a GPT☆211Updated 8 months ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆120Updated 9 months ago
- Flow-matching algorithms in JAX☆90Updated 8 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆103Updated 5 months ago
- σ-GPT: A New Approach to Autoregressive Models☆63Updated 8 months ago
- My take on Flow Matching☆52Updated 3 months ago
- ☆311Updated 2 years ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆137Updated 2 months ago
- ☆150Updated 8 months ago
- A JAX implementation of the continuous time formulation of Consistency Models☆84Updated 2 years ago
- 🧱 Modula software package☆188Updated last month
- Implementation of rectified flow and some of its followup research / improvements in Pytorch☆285Updated last week
- Simple and Effective Masked Diffusion Language Model☆376Updated 3 weeks ago
- Use Jax functions in Pytorch☆239Updated last year
- ☆95Updated last year
- ☆301Updated 10 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆271Updated 10 months ago
- Efficient optimizers☆190Updated this week
- Explorations into the recently proposed Taylor Series Linear Attention☆99Updated 8 months ago
- Minimal Implementation of a D3PM in pytorch☆216Updated last year