wmn-231314 / diffusion-data-constraintLinks
Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion models are significantly more data-efficient than standard left to right autoregressive models, due to their ability to learn from different token orderings.
☆120Updated last month
Alternatives and similar repositories for diffusion-data-constraint
Users that are interested in diffusion-data-constraint are comparing it to the libraries listed below
Sorting:
- The official github repo for "Diffusion Language Models are Super Data Learners".☆221Updated 3 months ago
- [ICML 2025] Roll the dice & look before you leap: Going beyond the creative limits of next-token prediction☆84Updated 8 months ago
- Official Code Repository for the paper "Continuous Diffusion Model for Language Modeling" (NeurIPS 2025).☆58Updated 4 months ago
- Official PyTorch Implementation for Vision-Language Models Create Cross-Modal Task Representations, ICML 2025☆32Updated 9 months ago
- Remasking Discrete Diffusion Models with Inference-Time Scaling☆66Updated this week
- ☆33Updated last year
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆137Updated last month
- [AAAI26] LongLLaDA: Unlocking Long Context Capabilities in Diffusion LLMs☆52Updated 2 months ago
- ☆34Updated 8 months ago
- [ICLR 2025] Source code for paper "A Spark of Vision-Language Intelligence: 2-Dimensional Autoregressive Transformer for Efficient Finegr…☆79Updated last year
- [ICLR 2026] RPG: KL-Regularized Policy Gradient (https://arxiv.org/abs/2505.17508)☆65Updated 2 weeks ago
- [NeurIPS-2024] 📈 Scaling Laws with Vocabulary: Larger Models Deserve Larger Vocabularies https://arxiv.org/abs/2407.13623☆89Updated last year
- ☆64Updated this week
- Optimizing Anytime Reasoning via Budget Relative Policy Optimization☆51Updated 6 months ago
- Multiplex Thinking: Reasoning via Token-wise Branch-and-Merge☆104Updated last week
- Unofficial Implementation of Selective Attention Transformer☆20Updated last year
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆89Updated last year
- ☆91Updated last year
- Esoteric Language Models☆111Updated this week
- Defeating the Training-Inference Mismatch via FP16☆181Updated 2 months ago
- [ICLR 2025] Official Pytorch Implementation of "Mix-LN: Unleashing the Power of Deeper Layers by Combining Pre-LN and Post-LN" by Pengxia…☆29Updated 6 months ago
- Code for "Reasoning to Learn from Latent Thoughts"☆124Updated 10 months ago
- The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling☆42Updated last month
- ☆111Updated 4 months ago
- Implementation of Negative-aware Finetuning (NFT) algorithm for "Bridging Supervised Learning and Reinforcement Learning in Math Reasonin…☆68Updated 5 months ago
- Stick-breaking attention☆62Updated 7 months ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆153Updated 11 months ago
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆56Updated 11 months ago
- [NeurIPS 2025] Official implementation for our paper "Scaling Diffusion Transformers Efficiently via μP".☆95Updated 3 months ago
- Code accompanying the paper "Generalized Interpolating Discrete Diffusion"☆113Updated 8 months ago