diff --git a/docs/cookbook/autoregressive/Qwen/Qwen3.5.mdx b/docs/cookbook/autoregressive/Qwen/Qwen3.5.mdx
index 30c92a678..11b10d0f0 100644
--- a/docs/cookbook/autoregressive/Qwen/Qwen3.5.mdx
+++ b/docs/cookbook/autoregressive/Qwen/Qwen3.5.mdx
@@ -157,7 +157,7 @@ This section provides deployment configurations optimized for different hardware
- **MI325X (256GB)** runs with tp=2.
- **MI355X (288GB)** runs with tp=2.
- **FP4**: The FP4 quantized model requires ~250GB for weights, cutting memory by almost 4x. NVFP4 ([nvidia/Qwen3.5-397B-A17B-NVFP4-V2](https://huggingface.co/nvidia/Qwen3.5-397B-A17B-NVFP4-V2)) requires B200/B300 (Blackwell architecture); AMD provides an MXFP4 checkpoint ([amd/Qwen3.5-397B-A17B-MXFP4](https://huggingface.co/amd/Qwen3.5-397B-A17B-MXFP4)) for MI355X.
- - **B200 (183GB)** runs with tp=4. (NVFP4)
+ - **B200 (183GB)** runs with tp=4 (tp=2 with expert parallelism 2 when MTP is enabled). (NVFP4)
- **B300 (275GB)** runs with tp=2. (NVFP4)
- **MI355X (288GB)** runs with tp=2 (use tp=4 for low concurrency). (MXFP4)
@@ -191,7 +191,7 @@ This section provides deployment configurations optimized for different hardware
183GB |
8 |
4 |
- 4 |
+ 4 / 2 + EP2 (MTP) |
| B300 |
diff --git a/docs/src/snippets/autoregressive/qwen35-deployment.jsx b/docs/src/snippets/autoregressive/qwen35-deployment.jsx
index 0ce212be0..4d1fbd4af 100644
--- a/docs/src/snippets/autoregressive/qwen35-deployment.jsx
+++ b/docs/src/snippets/autoregressive/qwen35-deployment.jsx
@@ -20,7 +20,7 @@ export const Qwen35Deployment = () => {
// 35B-A3B: H100 tp=1, H200 tp=1, B200 tp=1, B300 tp=1, MI300X tp=1, MI325X tp=1, MI355X tp=1
// 27B: tp=1 on all hardware (including MI300X, MI325X, MI355X)
//
- // FP4 (397B only): NVFP4 on Blackwell B200/B300 tp=4; AMD MXFP4 on MI355X tp=2
+ // FP4 (397B only): NVFP4 on Blackwell B200 tp=4 (tp=2 ep=2 w/ MTP) / B300 tp=4; AMD MXFP4 on MI355X tp=2
const MOE_MODELS = new Set(['397b', '122b', '35b']);
const FP8_MODELS = new Set(['397b', '122b', '35b', '27b']);
@@ -315,6 +315,11 @@ export const Qwen35Deployment = () => {
if (model === '122b' && hardware === 'h100' && quantization === 'fp8' && speculative === 'enabled') {
hwConfig = { ...hwConfig, tp: 4, mem: undefined };
}
+ // 397B B200 NVFP4 with MTP: tp=2 with expert parallelism 2 (TEP2) beats
+ // tp=4 across the concurrency sweep.
+ if (model === '397b' && hardware === 'b200' && quantization === 'fp4' && speculative === 'enabled') {
+ hwConfig = { ...hwConfig, tp: 2, ep: 2, mem: 0.8 };
+ }
let modelName;
if (quantization === 'fp4') {