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☆701Sep 29, 2025Updated 9 months ago
- [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆94Feb 14, 2025Updated last year
- A curated list for awesome discrete diffusion models resources.☆571Sep 9, 2025Updated 10 months ago
- Simple Guidance Mechanisms for Discrete Diffusion Models☆91Dec 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)☆739Feb 29, 2024Updated 2 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆72May 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☆109Feb 8, 2024Updated 2 years ago
- Official PyTorch implementation for ICLR2025 paper "Scaling up Masked Diffusion Models on Text"☆384Dec 22, 2024Updated last year
- The official implementation of "[MASK] is All You Need"☆127Jul 23, 2025Updated 11 months ago
- Remasking Discrete Diffusion Models with Inference-Time Scaling☆77Feb 7, 2026Updated 5 months ago
- Minimal (unofficial) implementation of the flow map matching method.☆15Jul 9, 2025Updated last year
- [ICLR 2025 Oral] Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models☆1,021Jul 10, 2025Updated last year
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Minimal Implementation of a D3PM in pytorch☆306Apr 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,905Jul 15, 2026Updated last week
- ☆18Sep 27, 2023Updated 2 years ago
- LLaDA implementation☆19Jul 24, 2025Updated 11 months ago
- [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☆138Aug 26, 2024Updated last year
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- ☆38Mar 29, 2025Updated last year
- 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 8 months ago
- ☆218Feb 12, 2024Updated 2 years ago
- [ICLR2025] DiffuGPT and DiffuLLaMA: Scaling Diffusion Language Models via Adaptation from Autoregressive Models☆400May 31, 2025Updated last year
- Reproduce ICLR2025 Energy-Based Diffusion Language Models for Text Generation☆69Jul 22, 2025Updated 11 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆68Mar 11, 2024Updated 2 years ago
- TorchCFM: a Conditional Flow Matching library☆2,535Updated this week
- Python package for P2 (Path Planning), a masked diffusion model sampling method for sequence generation (protein, text, etc.).☆23Aug 19, 2025Updated 11 months ago
- ☆15Aug 16, 2025Updated 11 months ago
- A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…☆4,625Jan 5, 2026Updated 6 months ago
- 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.
- Accelerate TarFlow Sampling with GS-Jacobi Iteration☆17Dec 13, 2025Updated 7 months 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☆847Dec 9, 2024Updated last year
- [ICLR 2022] Path integral sampler☆54Aug 21, 2023Updated 2 years ago
- Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"☆270Jan 17, 2025Updated last year
- [NeurIPS 2025 Spotlight] FUDOKI: Discrete Flow-based Unified Understanding and Generation via Kinetic-Optimal Velocities☆77Dec 21, 2025Updated 7 months ago