Official Jax Implementation of MD4 Masked Diffusion Models
☆155Feb 27, 2025Updated last year
Alternatives and similar repositories for md4
Users that are interested in md4 are comparing it to the libraries listed below
Sorting:
- [NeurIPS 2024] Simple and Effective Masked Diffusion Language Model☆629Sep 29, 2025Updated 5 months ago
- Code for the paper https://arxiv.org/abs/2402.04997☆105Feb 8, 2024Updated 2 years ago
- Implementation of Action Matching for the Schrödinger equation☆25Jun 18, 2023Updated 2 years ago
- [ICLR 2025 Oral] Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models☆963Jul 10, 2025Updated 7 months ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆64May 30, 2025Updated 9 months ago
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆707Feb 29, 2024Updated 2 years ago
- Official repository for "Plug & Play Directed Evolution for Proteins with Gradient-Based Discrete MCMC"☆13Jul 18, 2023Updated 2 years ago
- The official implementation of "[MASK] is All You Need"☆126Jul 23, 2025Updated 7 months ago
- [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆86Feb 14, 2025Updated last year
- [ICLR2025] DiffuGPT and DiffuLLaMA: Scaling Diffusion Language Models via Adaptation from Autoregressive Models☆369May 31, 2025Updated 9 months ago
- ☆36Mar 29, 2025Updated 11 months ago
- Official implementation of Metric Flow Matching (NeurIPS 2024)☆55Aug 14, 2025Updated 6 months ago
- ☆32May 30, 2024Updated last year
- ☆20May 28, 2025Updated 9 months ago
- Minimal Implementation of a D3PM in pytorch☆287Apr 22, 2024Updated last year
- ☆18Sep 27, 2023Updated 2 years ago
- A curated list for awesome discrete diffusion models resources.☆538Sep 9, 2025Updated 5 months ago
- Multiscale de-noising score matching☆20Feb 10, 2021Updated 5 years ago
- Reproduce ICLR2025 Energy-Based Diffusion Language Models for Text Generation☆65Jul 22, 2025Updated 7 months ago
- Educational implementation of the Discrete Flow Matching paper☆132Aug 26, 2024Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆67Mar 11, 2024Updated last year
- Scalable Minecraft multiplayer data collection engine☆70Updated this week
- Official PyTorch implementation for "Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data" (ICLR…☆82May 30, 2025Updated 9 months ago
- ☆10Dec 8, 2023Updated 2 years ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆46Dec 9, 2024Updated last year
- Python interface to cd-hit☆10Feb 26, 2019Updated 7 years ago
- ☆15May 29, 2025Updated 9 months ago
- Multiscale earth mover's distance methods for tree and manifold structured data☆12Mar 8, 2025Updated 11 months ago
- Trying to train LoRAs for ESM-2☆12Aug 6, 2023Updated 2 years ago
- ☆25Mar 26, 2024Updated last year
- Remasking Discrete Diffusion Models with Inference-Time Scaling☆67Feb 7, 2026Updated 3 weeks ago
- ☆19Jul 31, 2025Updated 7 months ago
- orbital MCMC☆10Jun 17, 2021Updated 4 years ago
- Library for epidemics on hypergraphs☆12May 13, 2024Updated last year
- [NeurIPS 2025] Beyond Masked and Unmasked: Discrete Diffusion Models via Partial Masking☆22Oct 22, 2025Updated 4 months ago
- ☆22Dec 19, 2025Updated 2 months ago
- [Neurips 2025] Official Implementation of ConfRover models☆20Feb 3, 2026Updated 3 weeks ago
- Official implementation of Deep Momentum Schrödinger Bridge☆30Nov 5, 2023Updated 2 years ago
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆4,167Jan 5, 2026Updated last month