kuleshov-group / remdmLinks
Remasking Discrete Diffusion Models with Inference-Time Scaling
☆65Updated 10 months ago
Alternatives and similar repositories for remdm
Users that are interested in remdm are comparing it to the libraries listed below
Sorting:
- Official Code Repository for the paper "Continuous Diffusion Model for Language Modeling" (NeurIPS 2025).☆56Updated 4 months ago
- Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion mod…☆119Updated 2 weeks ago
- Official PyTorch implementation for "Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data" (ICLR…☆78Updated 7 months ago
- ☆111Updated 2 years ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆152Updated 11 months ago
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆137Updated last month
- [ICLR2025] DiffuGPT and DiffuLLaMA: Scaling Diffusion Language Models via Adaptation from Autoregressive Models☆361Updated 7 months ago
- Implementation of "Reinforcing the Diffusion Chain of Lateral Thought with Diffusion Language Models" [NeurIPS 2025]☆71Updated last month
- [NeurIPS '25] Multi-Token Prediction Needs Registers☆26Updated last month
- [NeurIPS 2024] Code for the paper "Diffusion of Thoughts: Chain-of-Thought Reasoning in Diffusion Language Models"☆194Updated 10 months ago
- [NeurIPS 2025] Official implementation for our paper "Scaling Diffusion Transformers Efficiently via μP".☆94Updated 2 months ago
- Listing some diffusion papers in NLP domain I have read, text generation is main, table will continue to be updated.☆76Updated 10 months ago
- TraceRL & TraDo-8B: Revolutionizing Reinforcement Learning Framework for Diffusion Large Language Models☆398Updated last month
- Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]☆84Updated 9 months ago
- Diffusion Language Models For Code Infilling Beyond Fixed-size Canvas☆99Updated 4 months ago
- [AAAI26] LongLLaDA: Unlocking Long Context Capabilities in Diffusion LLMs☆51Updated last month
- Code for the paper: "Fine-Tuning Discrete Diffusion Models with Policy Gradient Methods"☆32Updated 8 months ago
- [NeurIPS'25] dKV-Cache: The Cache for Diffusion Language Models☆129Updated 8 months ago
- ☆19Updated last year
- ☆55Updated 7 months ago
- Implementation of Negative-aware Finetuning (NFT) algorithm for "Bridging Supervised Learning and Reinforcement Learning in Math Reasonin…☆67Updated 4 months ago
- Official PyTorch implementation for ICLR2025 paper "Scaling up Masked Diffusion Models on Text"☆360Updated last year
- Easy and Efficient dLLM Fine-Tuning☆203Updated last week
- Optimizing Anytime Reasoning via Budget Relative Policy Optimization☆51Updated 6 months ago
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆56Updated 10 months ago
- A collection of papers on discrete diffusion models☆167Updated 6 months ago
- The codebase of our paper "Improving the Training of Rectified Flows", NeurIPS 2024☆127Updated last year
- The this is the official implementation of "DAPE: Data-Adaptive Positional Encoding for Length Extrapolation"☆40Updated last year
- ☆50Updated last year
- Official PyTorch Implementation for Vision-Language Models Create Cross-Modal Task Representations, ICML 2025☆31Updated 8 months ago