google-deepmind / md4
Code for https://arxiv.org/abs/2406.04329
☆51Updated last month
Alternatives and similar repositories for md4:
Users that are interested in md4 are comparing it to the libraries listed below
- Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising"☆35Updated 2 months ago
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆26Updated this week
- Flow-matching algorithms in JAX☆82Updated 5 months ago
- Code repository for Trajectory Flow Matching☆50Updated 2 months ago
- Educational implementation of the Discrete Flow Matching paper☆71Updated 4 months ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆54Updated 4 months ago
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆51Updated 9 months ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆43Updated 4 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆48Updated 5 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- Simple Guidance Mechanisms for Discrete Diffusion Models☆17Updated last month
- ☆31Updated 7 months ago
- Simplified Masked Diffusion Language Model☆251Updated last month
- ☆91Updated last month
- ☆20Updated 3 months ago
- Code release for "Stochastic Optimal Control Matching"☆30Updated 5 months ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated 11 months ago
- Simple Scalable Discrete Diffusion for text in PyTorch☆29Updated 3 months ago
- A State-Space Model with Rational Transfer Function Representation.☆76Updated 8 months ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆126Updated last year
- ☆44Updated 8 months ago
- ☆95Updated 8 months ago
- Artificial Kuramoto Oscillatory Neurons☆45Updated this week
- Implementation of Particle Guidance: non-I.I.D. Diverse Sampling with Diffusion Models☆65Updated last year
- Official implementation of Metric Flow Matching (NeurIPS 2024)☆26Updated 2 months ago
- Implementing Controlled Monte Carlo Diffusions (ICLR 2024)☆14Updated 3 months ago
- Annotated Flow Matching paper☆157Updated 4 months ago
- Implementation of papers in 101 lines of code.☆18Updated last year
- Explorations into the recently proposed Taylor Series Linear Attention☆91Updated 4 months ago
- ☆50Updated 3 months ago