Official Jax Implementation of MD4 Masked Diffusion Models
☆163Feb 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☆719Sep 29, 2025Updated last year
- [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆100Feb 14, 2025Updated last year
- A curated list for awesome discrete diffusion models resources.☆579Sep 9, 2025Updated last year
- Simple Guidance Mechanisms for Discrete Diffusion Models☆93Dec 16, 2024Updated 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
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- 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]☆85Apr 24, 2025Updated last year
- Code for the paper https://arxiv.org/abs/2402.04997☆114Feb 8, 2024Updated 2 years ago
- 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"☆16Mar 26, 2025Updated last year
- Remasking Discrete Diffusion Models with Inference-Time Scaling☆78Feb 7, 2026Updated 8 months ago
- Minimal (unofficial) implementation of the flow map matching method.☆16Jul 9, 2025Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- [ICLR 2025 Oral] Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models☆1,036Jul 10, 2025Updated last year
- Minimal Implementation of a D3PM in pytorch☆310Apr 22, 2024Updated 2 years ago
- Implementation of Action Matching for the Schrödinger equation☆25Jun 18, 2023Updated 3 years ago
- Official PyTorch implementation for "Large Language Diffusion Models"☆3,994Jul 15, 2026Updated 2 months ago
- ☆18Sep 27, 2023Updated 3 years ago
- LLaDA implementation☆19Jul 24, 2025Updated last year
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆51Dec 9, 2024Updated 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
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Reviving Any-Order Autoregressive Models via Principled Parallel Sampling and Speculative Decoding☆16Nov 16, 2025Updated 10 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
- ☆39Aug 30, 2026Updated last month
- Reproduce ICLR2025 Energy-Based Diffusion Language Models for Text Generation☆71Jul 22, 2025Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆70Mar 11, 2024Updated 2 years ago
- Python package for P2 (Path Planning), a masked diffusion model sampling method for sequence generation (protein, text, etc.).☆26Aug 19, 2025Updated last year
- ☆16Aug 16, 2025Updated last year
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆4,798Jan 5, 2026Updated 9 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.
- Accelerate TarFlow Sampling with GS-Jacobi Iteration☆17Dec 13, 2025Updated 9 months ago
- Code for "Evaluating Spatial Understanding of Large Language Models" TMLR 2024.☆16Feb 22, 2024Updated 2 years ago
- Official Repository for "Unbalancedness in Neural Monge Maps Improves Unpaired Domain Translation" [ICLR 2024]☆18May 15, 2024Updated 2 years ago
- Gradient Estimation with Discrete Stein Operators (NeurIPS 2022)☆18Nov 14, 2023Updated 2 years ago
- EDM2 and Autoguidance -- Official PyTorch implementation☆853Dec 9, 2024Updated last year
- [ICLR 2022] Path integral sampler☆55Aug 21, 2023Updated 3 years ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆273Jan 17, 2025Updated last year