chen-hao-chao / mdm-primeLinks
[NeurIPS 2025] Beyond Masked and Unmasked: Discrete Diffusion Models via Partial Masking
☆20Updated last month
Alternatives and similar repositories for mdm-prime
Users that are interested in mdm-prime are comparing it to the libraries listed below
Sorting:
- A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.☆190Updated 3 months ago
- ☆60Updated 2 months ago
- Official implementation for our paper "Scaling Diffusion Transformers Efficiently via μP".☆90Updated 3 months ago
- UniDisc: A discrete diffusion model for joint multimodal generation, enabling controllable and efficient text-image synthesis, editing, a…☆128Updated 6 months ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆132Updated 7 months ago
- Official PyTorch Implementation for Dual-Process Image Generation, ICCV 2025☆99Updated last month
- ☆29Updated 5 months ago
- Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]☆80Updated 5 months ago
- ☆20Updated 3 months ago
- Official PyTorch implementation of TokenSet.☆125Updated 6 months ago
- [NeurIPS 2025, Spotlight]: Ambient-o: Training Good models with Bad Data.☆28Updated 2 months ago
- Flash Attention Triton kernel with support for second-order derivatives☆101Updated last week
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆44Updated 4 months ago
- ☆115Updated last week
- RS-IMLE☆42Updated 10 months ago
- [ICCV 2025] Official Implementation of Contrastive Flow Matching☆129Updated 3 months ago
- Inference-Time Alignment in Protein Diffusion Models☆44Updated 8 months ago
- Code for the paper DisCo-Diff: Enhancing Continuous Diffusion Models with Discrete Latents, ICML 2024☆89Updated last year
- [Preprint] UCGM: Unified Continuous Generative Models☆168Updated 4 months ago
- FlexTok: Resampling Images into 1D Token Sequences of Flexible Length☆255Updated 4 months ago
- [ICML 2024] Compositional Image Decomposition with Diffusion Models☆51Updated last year
- NeuMeta transforms neural networks by allowing a single model to adapt on the fly to different sizes, generating the right weights when n…☆43Updated 11 months ago
- A Video Tokenizer Evaluation Dataset☆135Updated 9 months ago
- [ICML 2025] Gaussian Mixture Flow Matching Models (GMFlow)☆139Updated 4 months ago
- Official Code Repository for the paper "Continuous Diffusion Model for Language Modeling" (NeurIPS 2025).☆44Updated 3 weeks ago
- [NeurIPS 2025 Oral] Exploring Diffusion Transformer Designs via Grafting☆60Updated 4 months ago
- Official implementation for Diffusion Alignment as Sampling (DAS), ICLR'25, Spotlight☆51Updated 8 months ago
- Simple Guidance Mechanisms for Discrete Diffusion Models☆55Updated 10 months ago
- Minimal Implementation of a D3PM in pytorch☆267Updated last year
- Awesome List of Consistency Models☆117Updated 5 months ago