GitHub - NeelM47/mdm-adaptive-inference: PyTorch implementation of "Train for the Worst, Plan for the Best." Investigating adaptive token ordering in Masked Diffusion Models (MDMs) to sidestep hard subproblems and elicit reasoning in discrete domains.

Code implementation of the paper [ https://arxiv.org/pdf/2502.06768 ]

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
.gitignore		.gitignore
README.md		README.md
requirements.txt		requirements.txt

Provide feedback