Views
No views yet
train_sft split of HuggingFaceH4/ultrachat_200k. Responses were regenerated by Qwen3.6-35B-A3B. DSpark extends DFlash with a Markov head (intra-block token dependency) and a confidence head (per-position acceptance prediction). Training compute for this model was generously provided by Lambda, a leading cloud platform for AI training and inference.1# In virtual environment with speculators installed
2python scripts/prepare_data.py \
3 --model Qwen/Qwen3.6-35B-A3B \
4 --data ./data/qwen3_6_35b_a3b_regen/magpie_output.jsonl \
5 --data ./data/qwen3_6_35b_a3b_regen/ultrachat_output.jsonl \
6 --output ./output/dspark_qwen3_6_35b_a3b \
7 --seq-length 160001# In (separate) virtual environment with vllm installed
2CUDA_VISIBLE_DEVICES=0,1,2,3 python scripts/launch_vllm.py \
3 Qwen/Qwen3.6-35B-A3B \
4 --target-layer-ids 2 10 20 30 37 \
5 -- --port 8000 \
6 --tensor-parallel-size 4 \
7 --max-model-len 16002 \
8 --enforce-eager1# In virtual environment with speculators installed
2CUDA_VISIBLE_DEVICES=4,5,6,7 torchrun \
3 --standalone \
4 --nproc_per_node 4 \
5 scripts/train.py \
6 --verifier-name-or-path Qwen/Qwen3.6-35B-A3B \
7 --speculator-type dspark \
8 --data-path ./output/dspark_qwen3_6_35b_a3b \
9 --vllm-endpoint http://localhost:8000/v1 \
10 --save-path ./output/dspark_qwen3_6_35b_a3b/checkpoints \
11 --block-size 8 \
12 --max-anchors 3072 \
13 --target-layer-ids 2 10 20 30 37 \
14 --num-layers 5 \
15 --markov-rank 256 \
16 --markov-head-type vanilla \
17 --enable-confidence-head \
18 --confidence-head-with-markov \
19 --loss-fn '{"ce": 0.1, "tv": 0.9}' \
20 --confidence-head-alpha 1.0 \
21 --epochs 5 \
22 --lr 3e-4 \
23 --total-seq-len 16000 \
24 --on-missing generate \
25 --on-generate delete \
26 --checkpoint-freq 0.1| Base Model | Qwen/Qwen3.6-35B-A3B |
| Chat Template | Qwen/Qwen3.6-35B-A3B (use /chat/completions endpoint) |
| Format | Safetensors |
| License | Apache 2.0 |
| Validation Hardware | Nvidia H100 |
1# Deploy with speculative decoding
2vllm serve Qwen/Qwen3.6-35B-A3B \
3 --tensor-parallel-size 2 \
4 --speculative-config '{
5 "model": "RedHatAI/Qwen3.6-35B-A3B-speculator.dspark",
6 "num_speculative_tokens": 8,
7 "method": "dspark"
8 }'| Dataset | Pos 0 | Pos 1 | Pos 2 | Pos 3 | Pos 4 | Pos 5 | Pos 6 | Pos 7 | Avg. Length |
|---|---|---|---|---|---|---|---|---|---|
| HumanEval | 82.0% | 66.2% | 54.1% | 44.1% | 36.4% | 29.9% | 24.7% | 20.4% | 4.58 |
| math_reasoning | 84.0% | 70.3% | 59.8% | 51.2% | 43.5% | 37.0% | 31.2% | 26.5% | 5.03 |
| qa | 71.8% | 50.9% | 37.4% | 28.0% | 20.9% | 16.0% | 12.1% | 9.2% | 3.46 |
| question | 73.5% | 53.1% | 39.5% | 30.4% | 24.2% | 19.5% | 15.6% | 12.7% | 3.68 |
| rag | 78.2% | 57.1% | 44.1% | 34.7% | 27.3% | 21.2% | 16.4% | 13.1% | 3.92 |
| summarization | 74.2% | 53.6% | 40.4% | 30.6% | 23.5% | 17.8% | 13.7% | 10.5% | 3.64 |
| tool_call | 71.7% | 50.8% | 36.1% | 26.3% | 19.5% | 14.7% | 11.2% | 8.6% | 3.39 |
| translation | 70.7% | 51.2% | 37.8% | 28.4% | 21.2% | 15.7% | 11.9% | 8.8% | 3.46 |
| writing | 74.4% | 54.3% | 40.8% | 31.6% | 25.2% | 20.3% | 16.5% | 13.4% | 3.76 |