Official Jax Implementation of MD4 Masked Diffusion Models
☆164Feb 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☆724Sep 29, 2025Updated last year
- [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆100Feb 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)☆747Feb 29, 2024Updated 2 years ago
- Simple Guidance Mechanisms for Discrete Diffusion Models☆93Dec 16, 2024Updated last year
- A curated list for awesome discrete diffusion models resources.☆578Sep 9, 2025Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Code for the paper: "Generalized Discrete Diffusion from Snapshots"☆14Apr 2, 2026Updated 6 months ago
- Pytorch Implementation of MD4: Simplified and Generalized Masked Diffusion for Discrete Data☆13Feb 5, 2025Updated last year
- [ICML 2025] The Diffusion Duality☆245Jun 6, 2026Updated 4 months ago
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆76May 30, 2025Updated last year
- Official PyTorch implementation for "Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data" (ICLR…☆84May 30, 2025Updated last year
- Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]☆86Apr 24, 2025Updated last year
- Official PyTorch implementation for ICLR2025 paper "Scaling up Masked Diffusion Models on Text"☆389Dec 22, 2024Updated last year
- The official implementation of "[MASK] is All You Need"☆126Jul 23, 2025Updated last year
- Official implementation of NeurIPS'24 paper "Causal Language Modeling Can Elicit Search and Reasoning Capabilities on Logic Puzzles"☆17Mar 26, 2025Updated last year
- 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.
- Remasking Discrete Diffusion Models with Inference-Time Scaling☆79Feb 7, 2026Updated 8 months ago
- [ICLR 2025 Oral] Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models☆1,039Jul 10, 2025Updated last year
- Minimal (unofficial) implementation of the flow map matching method.☆16Jul 9, 2025Updated last year
- Code for the paper https://arxiv.org/abs/2402.04997☆115Feb 8, 2024Updated 2 years ago
- ☆116May 29, 2023Updated 3 years ago
- Official PyTorch implementation for "Large Language Diffusion Models"☆4,012Jul 15, 2026Updated 3 months ago
- Minimal Implementation of a D3PM in pytorch☆310Apr 22, 2024Updated 2 years ago
- Collecting research materials on neural samplers with diffusion/flow models☆60Jul 4, 2025Updated last year
- ☆18Sep 27, 2023Updated 3 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Reproduce ICLR2025 Energy-Based Diffusion Language Models for Text Generation☆71Jul 22, 2025Updated last year
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆51Dec 9, 2024Updated last year
- LightningDiT Fork with TREAD implementation☆13Oct 13, 2025Updated last year
- Implementation of Action Matching for the Schrödinger equation☆25Jun 18, 2023Updated 3 years ago
- Reviving Any-Order Autoregressive Models via Principled Parallel Sampling and Speculative Decoding☆16Nov 16, 2025Updated 11 months ago
- ☆221Feb 12, 2024Updated 2 years ago
- [ICLR2025] DiffuGPT and DiffuLLaMA: Scaling Diffusion Language Models via Adaptation from Autoregressive Models☆406May 31, 2025Updated last year
- ☆144May 7, 2024Updated 2 years ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆854Dec 9, 2024Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆56Mar 10, 2025Updated last year
- LLaDA implementation☆19Jul 24, 2025Updated last year
- Educational implementation of the Discrete Flow Matching paper☆142Aug 26, 2024Updated 2 years ago
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- ☆40Aug 30, 2026Updated 2 months ago
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆4,824Jan 5, 2026Updated 9 months ago
- ☆16Aug 16, 2025Updated last year