Repository navigation
feat(slurm): support external model providers - #1080
Draft
dhruvnathawani wants to merge 4 commits into
Draft
dhruvnathawani wants to merge 4 commits into
dhruvnathawani wants to merge 4 commits into
Conversation
Allow hosted endpoints alongside managed models and run hosted-only recipes in CPU allocations. Resolve credentials at runtime and preserve native Data Designer model settings and asynchronous execution. Reuse shard, retry, resume, and collection contracts while preserving existing local-only plan serialization. Add configuration examples, documentation, and native client and allocation lifecycle coverage. Closes #1079 Signed-off-by: Dhruv Nathawani <128275431+dhruvnathawani@users.noreply.github.com>
Signed-off-by: Dhruv Nathawani <128275431+dhruvnathawani@users.noreply.github.com>
Contributor
|
All contributors have signed the DCO ✍️ ✅ |
Contributor
|
Fern preview: https://nvidia-preview-pr-1080.docs.buildwithfern.com/nemo/datadesigner
|
dhruvnathawani
force-pushed
the
dhruv/slurm-hosted-providers
branch
from
October 23, 2026 05:52
2f19916 to
dfcbbf0
Compare
Signed-off-by: Dhruv Nathawani <128275431+dhruvnathawani@users.noreply.github.com>
Signed-off-by: Dhruv Nathawani <128275431+dhruvnathawani@users.noreply.github.com>
2 tasks done
This branch has not been deployed
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
📋 Summary
Allow Slurm recipes to use hosted model endpoints alongside managed vLLM models, including hosted-only recipes in CPU allocations. Existing DD async execution remains unchanged.
Targets
feat/slurm-execution(integration PR #1040). Keep draft until mixed-provider live qualification is complete.🔗 Related Issue
Closes #1079
🔄 Changes
--gres=noneand skip GPU discovery, serving images, containers, and proxies.submission.memoryfor generation and collection; an explicit value overrides GPU memory requests for generation.Async execution does not guarantee continuous GPU activity. Representative utilization and cluster policy remain prerequisites for qualifying the SQL migration.
🧪 Testing
make testacross all four packages: 6,531 passed, 3 skipped.make test-slurmincluding live-run regressions: 2,019 passed.make check-all, Slurm Ruff checks, andgit diff --checkpassed.make check-fern-linkspassed across 101 pages; full Fern validation and all visible PR checks passed on the final head.pass, successful attempt and winner publication, then successful winner-driven collection of the same two rows. Generation and collection each allocated four CPUs, 16 GB, and zero GPUs. Persisted text artifacts passed the hosted-credential scan.qwen4_expand the image lacksQwen4ExpForConditionalGeneration. No GPU allocation or utilization measurement has occurred.✅ Checklist