Skip to content

feat(draft): stream vocab-parallel soft CE #3703

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
terrykong merged 22 commits into from
Sep 17, 2026
Merged

feat(draft): stream vocab-parallel soft CE #3703

Show file tree
Hide file tree
Changes from 6 commits
Commits
Show all changes
22 commits
Select commit Hold shift + click to select a range
5239101
feat(draft): stream vocab-parallel soft CE
seonjinn Aug 18, 2026
1e3d8eb
test(draft): cover cached single-tile soft CE
seonjinn Aug 18, 2026
9cdc84f
perf(draft): cache single-tile soft CE
seonjinn Aug 19, 2026
5133c28
Merge branch 'main' into seonjinn/draft-cotrain-streaming-loss-20260818
seonjinn Aug 20, 2026
8887579
test(draft): assert streaming loss normalization shape
seonjinn Aug 22, 2026
895e0c2
Validate caller-supplied bin_ids range in streaming soft CE
seonjinn Aug 22, 2026
064fcf9
Merge remote-tracking branch 'origin/main' into review/pr-3703-main-m…
seonjinn Sep 2, 2026
a141bfa
Merge remote-tracking branch 'origin/main' into review/pr-3703-main-m…
seonjinn Sep 3, 2026
cbda213
Merge remote-tracking branch 'origin/main' into review/pr-3703-main-m…
seonjinn Sep 4, 2026
3e4cc82
feat(draft): expose draft soft-CE token_chunk_size as config
seonjinn Sep 4, 2026
f633cab
Fix import ordering in loss wrapper
seonjinn Sep 5, 2026
d9068ad
Add draft.token_chunk_size to grpo reference config
seonjinn Sep 7, 2026
e9c1c25
Merge remote-tracking branch 'origin/main' into review/pr-3703-main-m…
seonjinn Sep 10, 2026
fa49016
Merge remote-tracking branch 'origin/main' into review/pr-3703-main-m…
seonjinn Sep 10, 2026
122af10
Merge branch 'main' into seonjinn/draft-cotrain-streaming-loss-20260818
seonjinn Sep 12, 2026
4c6b332
Merge branch 'main' into PR #3703
seonjinn Sep 12, 2026
02b570f
Merge remote-tracking branch 'origin/main' into review/pr-3703-remote…
seonjinn Sep 15, 2026
093a351
Merge remote-tracking branch 'origin/main' into review/pr-3703-remote…
seonjinn Sep 16, 2026
7ff44d0
Fix broken NATIVE_MULTIMODAL_KEYS import in rollout manager
seonjinn Sep 16, 2026
128fbaa
Merge remote-tracking branch 'origin/main' into review/pr-3703-remote…
seonjinn Sep 16, 2026
7de1a99
Merge branch 'main' into seonjinn/draft-cotrain-streaming-loss-20260818
seonjinn Sep 17, 2026
9ca803f
Merge branch 'main' into seonjinn/draft-cotrain-streaming-loss-20260818
seonjinn Sep 17, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading
Loading