Official Jax Implementation of MD4 Masked Diffusion Models
☆161Feb 27, 2025Updated last year
Alternatives and similar repositories for md4
Users that are interested in md4 are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- [NeurIPS 2024] Simple and Effective Masked Diffusion Language Model☆698Sep 29, 2025Updated 9 months ago
- Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]☆85Apr 24, 2025Updated last year
- [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆94Feb 14, 2025Updated last year
- [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)☆734Feb 29, 2024Updated 2 years ago
- [ICLR 2025 Oral] Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models☆1,013Jul 10, 2025Updated 11 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- ☆38Mar 29, 2025Updated last year
- Implementation of Action Matching for the Schrödinger equation☆25Jun 18, 2023Updated 3 years ago
- [ICLR2025] DiffuGPT and DiffuLLaMA: Scaling Diffusion Language Models via Adaptation from Autoregressive Models☆395May 31, 2025Updated last year
- Minimal Implementation of a D3PM in pytorch☆304Apr 22, 2024Updated 2 years ago
- Official PyTorch implementation for "Large Language Diffusion Models"☆3,842Jun 25, 2026Updated last week
- Code for the paper https://arxiv.org/abs/2402.04997☆109Feb 8, 2024Updated 2 years ago
- Official PyTorch implementation for "Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data" (ICLR…☆83May 30, 2025Updated last year
- [NeurIPS 2025] Beyond Masked and Unmasked: Discrete Diffusion Models via Partial Masking☆31Jun 15, 2026Updated 2 weeks ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆71May 30, 2025Updated last year
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- The official implementation of "[MASK] is All You Need"☆126Jul 23, 2025Updated 11 months ago
- Official repository for "Plug & Play Directed Evolution for Proteins with Gradient-Based Discrete MCMC"☆12Jul 18, 2023Updated 2 years ago
- ☆32May 30, 2024Updated 2 years ago
- Remasking Discrete Diffusion Models with Inference-Time Scaling☆77Feb 7, 2026Updated 4 months ago
- A curated list for awesome discrete diffusion models resources.☆568Sep 9, 2025Updated 9 months ago
- Multiscale de-noising score matching☆20Feb 10, 2021Updated 5 years ago
- Official implementation of Metric Flow Matching (NeurIPS 2024)☆60Aug 14, 2025Updated 10 months ago
- Educational implementation of the Discrete Flow Matching paper☆136Aug 26, 2024Updated last year
- ☆18Sep 27, 2023Updated 2 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- ☆20May 28, 2025Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆68Mar 11, 2024Updated 2 years ago
- Gradient Estimation with Discrete Stein Operators (NeurIPS 2022)☆18Nov 14, 2023Updated 2 years ago
- Trying to train LoRAs for ESM-2☆13Aug 6, 2023Updated 2 years ago
- Reproduce ICLR2025 Energy-Based Diffusion Language Models for Text Generation☆69Jul 22, 2025Updated 11 months ago
- ☆93Jun 14, 2024Updated 2 years ago
- ☆57Apr 7, 2026Updated 2 months ago
- ☆33Jan 6, 2025Updated last year
- ☆10Jun 12, 2021Updated 5 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- ☆25Mar 26, 2024Updated 2 years ago
- This repository contains an experimental PyTorch implementation exploring the NoProp algorithm, presented in the paper "NOPROP: TRAINING …☆18Jun 20, 2026Updated last week
- ☆11Dec 8, 2023Updated 2 years ago
- code for "Riemannian Flow Matching on General Geometries".☆309Mar 13, 2024Updated 2 years ago
- Official implementation of Categorical Flow Maps on text.☆60Feb 16, 2026Updated 4 months ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆51Dec 9, 2024Updated last year
- [ICLR 2022] Path integral sampler☆54Aug 21, 2023Updated 2 years ago