SIGN IN SIGN UP

feat(dllm): add SCDD training and sampling (#3354)

* feat(dllm): add mixed mask+uniform corruption kernel

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* feat(dllm): add SCDD schedule and NELBO loss

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* feat(dllm): add scdd training strategy

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* feat(dllm): add SCDD sampler

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* docs(dllm): document SCDD fine-tuning

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* test(dllm): add cpu smoke for scdd training path

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* test(dllm): add single-gpu scdd smoke test

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* docs(dllm): use openreview link, fix scdd train command

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* test(dllm): use llama fixture for scdd smoke

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* perf(dllm): chunk the SCDD vocab reduction

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* feat(dllm): make llada_scdd the go-to SCDD recipe

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* test(dllm): cover scdd sampler edge branches

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* docs(dllm): use uv run for dllm commands

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* Update docs/guides/dllm/finetune.mdx

Co-authored-by: jgerh <163925524+jgerh@users.noreply.github.com>
Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* Update docs/guides/dllm/finetune.mdx

Co-authored-by: jgerh <163925524+jgerh@users.noreply.github.com>
Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* Update docs/guides/dllm/finetune.mdx

Co-authored-by: jgerh <163925524+jgerh@users.noreply.github.com>
Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* Update docs/guides/dllm/finetune.mdx

Co-authored-by: jgerh <163925524+jgerh@users.noreply.github.com>
Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* Update docs/guides/dllm/finetune.mdx

Co-authored-by: jgerh <163925524+jgerh@users.noreply.github.com>
Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* Update docs/guides/dllm/finetune.mdx

Co-authored-by: jgerh <163925524+jgerh@users.noreply.github.com>
Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* Update docs/guides/dllm/finetune.mdx

Co-authored-by: jgerh <163925524+jgerh@users.noreply.github.com>
Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* fix(dllm): validate scdd corruption bounds and keep t on the grid

check mask_prob + uniform_prob <= 1 in corrupt_mix, clamp the uniform
draw so a clean [MASK] cannot produce an out-of-range id, and snap the
top diffusion time to the previous grid point instead of 1 - 1e-4.

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* docs(dllm): fix scdd sampler wording on remasking

visible tokens are never sent back to [MASK]; the reverse posterior puts
no mass there. reported by the scdd authors.

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* docs(dllm): drop external repo paths from scdd provenance notes

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* test(dllm): align scdd smoke script and config with repo conventions

shebang on the first line, wrap pytest in coverage run like the other L2
scripts, and use distributed.strategy instead of the FSDP2Manager target.

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* docs(dllm): drop duplicated idlm config line

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* fix(dllm): scdd final denoise removes uniform-channel noise, not just masks

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

* pin sdpa in the scdd smoke, fp32 breaks flash attn

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>

---------

Signed-off-by: Kashif Rasul <kashif.rasul@gmail.com>
Co-authored-by: jgerh <163925524+jgerh@users.noreply.github.com>
K
Kashif Rasul committed
14da9dc2413f4cd28c9ac8e51083a32f36d3e999
Parent: 3d647a2
Committed by GitHub <noreply@github.com> on 8/24/2026, 5:59:51 PM