Skip to content

build(speechlm2): add cut-cross-entropy to compiled extras - #16241

Open
pzelasko wants to merge 8 commits into
mainfrom
codex/pr16158-rewrite/07-cut-cross-entropy-dependency
Open

build(speechlm2): add cut-cross-entropy to compiled extras#16241
pzelasko wants to merge 8 commits into
mainfrom
codex/pr16158-rewrite/07-cut-cross-entropy-dependency

Conversation

@pzelasko

@pzelasko pzelasko commented Sep 11, 2026

Copy link
Copy Markdown
Collaborator

Summary

This is stack 7/7. It isolates approval of the optional cut-cross-entropy dependency from the functional SpeechLM2 changes in the preceding PRs.

  • add cut-cross-entropy==25.3.2 to the compiled and compiled-a100 extras
  • pin apple/ml-cross-entropy to revision 87a86aba72cfd2f0d8abecaf81c13c4528ea07d8
  • refresh uv.lock

The earlier PRs may use cut-cross-entropy through optional paths, but do not declare or install it. This PR contains dependency metadata only.

Stack

  1. feat(speechlm2): add generalized speech encoder foundations #16192 — encoder foundations
  2. feat(data): add indexed resumable multimodal loading and exact token budgets #16193 — indexed and resumable multimodal data
  3. feat(speechlm2): add padding-free packed speech training #16194 — packed SpeechLM2 training
  4. feat(speechlm2): export portable checkpoints and serve them with vLLM #16195 — portable export and vLLM serving
  5. feat(speechlm2): add native MTP speculative decoding in vLLM #16196 — MTP serving
  6. Add DFlash speculative decoding support to NeMo SpeechLM vLLM plugin #16239 — DFlash serving
  7. this PR — cut-cross-entropy dependency

Base: #16239

Verification

  • uv lock --check --offline
  • repository pre-commit checks over this PR range
  • CCE-available and CCE-unavailable loss-selection tests
  • verified this PR changes only pyproject.toml and uv.lock
  • verified this dependency-only patch is identical to its pre-rebase version

No documentation change is needed because this PR only moves an existing optional dependency declaration into its own review unit.

@pzelasko
pzelasko requested a review from a team as a code owner September 11, 2026 16:28
@copy-pr-bot

copy-pr-bot Bot commented Sep 11, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@pzelasko

Copy link
Copy Markdown
Collaborator Author

/ok to test 01716cf

@pzelasko
pzelasko force-pushed the speech-pr/pr-16129-on-pr16196 branch from 2e41db0 to d0d99aa Compare September 11, 2026 21:26
@pzelasko
pzelasko force-pushed the codex/pr16158-rewrite/07-cut-cross-entropy-dependency branch from 01716cf to 159ec05 Compare September 11, 2026 21:26
@pzelasko
pzelasko added this pull request to stack #16243 September 11, 2026 21:26
@pzelasko

Copy link
Copy Markdown
Collaborator Author

/ok to test 159ec05

@pzelasko
pzelasko removed this pull request from stack #16243 September 12, 2026 00:57
Signed-off-by: slyne deng <slyned@nvidia.com>
Signed-off-by: slyne deng <slyned@nvidia.com>
Signed-off-by: slyne deng <slyned@nvidia.com>
Signed-off-by: slyne deng <slyned@nvidia.com>
Signed-off-by: slyne deng <slyned@nvidia.com>
Signed-off-by: SlyneD <slyned@nvidia.com>
Slyne and others added 2 commits September 12, 2026 01:02
Signed-off-by: SlyneD <slyned@nvidia.com>
Original-Commit: 2c645f4
Signed-off-by: Piotr Żelasko <pzelasko@nvidia.com>
@pzelasko
pzelasko force-pushed the speech-pr/pr-16129-on-pr16196 branch from d0d99aa to 2f5c05d Compare September 12, 2026 01:03
@pzelasko
pzelasko force-pushed the codex/pr16158-rewrite/07-cut-cross-entropy-dependency branch from 159ec05 to c36d384 Compare September 12, 2026 01:03
@pzelasko
pzelasko added this pull request to stack #16244 September 12, 2026 01:03
@pzelasko

Copy link
Copy Markdown
Collaborator Author

/ok to test c36d384

@pzelasko
pzelasko removed this pull request from stack #16244 September 12, 2026 01:08
Base automatically changed from speech-pr/pr-16129-on-pr16196 to main September 12, 2026 03:04
@github-actions

Copy link
Copy Markdown
Contributor

[🤖]: Hi @pzelasko 👋,

We wanted to let you know that a CICD pipeline for this PR just finished successfully.

So it might be time to merge this PR or get some approvals.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants