diff --git a/miles/utils/audit_utils/checksum_utils.py b/miles/utils/audit_utils/checksum_utils.py
index 64a5f1e1f0..6b6fe8d8d1 100644
--- a/miles/utils/audit_utils/checksum_utils.py
+++ b/miles/utils/audit_utils/checksum_utils.py
@@ -14,7 +14,7 @@ def flatten_inference_engine_checksums(check_weights_result: Any) -> list[Infere
 
 
 def _flatten_to_inference_engine_bodies(check_weights_result: Any) -> list[Any]:
-    return [engine_body for server in check_weights_result for server_group in server for engine_body in server_group]
+    return [engine_body for server_group in check_weights_result for engine_body in server_group]
 
 
 def _merge_inference_engine_ranks(engine_body: dict[str, Any]) -> InferenceEngineChecksums:
@@ -24,11 +24,18 @@ def _merge_inference_engine_ranks(engine_body: dict[str, Any]) -> InferenceEngin
     ranks: list[dict[str, Any]] = engine_body.get("ranks", []) or []
     assert ranks, f"check_weights engine body has no ranks: {engine_body!r}"
 
-    ranks_sorted = sorted(ranks, key=lambda r: r["parallelism_info"]["rank"])
+    ranks_sorted = sorted(ranks, key=_gpu_rank)
 
     merged: InferenceEngineChecksums = {}
     for rank_info in ranks_sorted:
-        rank = rank_info["parallelism_info"]["rank"]
+        rank = _gpu_rank(rank_info)
         for name, value in rank_info["checksums"].items():
             merged[f"rank{rank}/{name}"] = value
     return merged
+
+
+def _gpu_rank(rank_info: dict[str, Any]) -> int:
+    parallelism_info = rank_info["parallelism_info"]
+    gpu_ranks = {role_info["rank"] for role_info in parallelism_info}
+    assert len(gpu_ranks) == 1, f"expected one GPU rank across roles, got {gpu_ranks}: {rank_info!r}"
+    return next(iter(gpu_ranks))
diff --git a/tests/e2e/ft/README.md b/tests/e2e/ft/README.md
index 20d637284d..9d0413a5d0 100644
--- a/tests/e2e/ft/README.md
+++ b/tests/e2e/ft/README.md
@@ -3,7 +3,7 @@
 ## Layout
 
 - Scenario logic lives in `conftest_ft/scenario_<name>.py`.
-- CI runs it via thin per-mode entry files `test_trainer_ft_<scenario>_<mode>.py`, each registered with `register_cuda_ci(est_time=..., suite="stage-c-8-gpu-h100", labels=["ft-short"])` (comparison scenarios) or `labels=["ft-long"]` (soak scenarios).
+- CI runs it via thin per-mode entry files `test_trainer_ft_<scenario>_<mode>.py`, each registered with `register_cuda_ci(est_time=..., suite="stage-c-8-gpu-h200", labels=["ft-short"])` (comparison scenarios) or `labels=["ft-long"]` (soak scenarios).
 - The CUDA CI runner executes each entry as bare `python3 <file>` (exit code = pass/fail); the entry just calls the scenario's `run_ci(mode)`.
 
 | Scenario (`conftest_ft/scenario_*.py`) | Type | What it verifies |
@@ -36,7 +36,7 @@
 
 ### In CI
 
-- Gated on the `run-ci-ft-short` / `run-ci-ft-long` PR labels (FT is expensive — not run on every PR). `ft-short` covers the comparison scenarios (no_failure / deterministic / with_failure, minutes each); `ft-long` covers the soak scenarios (random-crash survival, realistic-gsm8k convergence — tens of minutes to hours). With a label set, the matching entries run on `stage-c-8-gpu-h100`.
+- Gated on the `run-ci-ft-short` / `run-ci-ft-long` PR labels (FT is expensive — not run on every PR). `ft-short` covers the comparison scenarios (no_failure / deterministic / with_failure, minutes each); `ft-long` covers the soak scenarios (random-crash survival, realistic-gsm8k convergence — tens of minutes to hours). With a label set, the matching entries run on `stage-c-8-gpu-h200`.
 - Add a `(scenario, mode)` to CI: copy an entry file, change `run_ci(...)`'s mode.
 - Add a new label: edit `tests/ci/labels.py` and create the matching `run-ci-<label>` GitHub label.
 
diff --git a/tests/e2e/ft/conftest_ft/app.py b/tests/e2e/ft/conftest_ft/app.py
index a25406c6e9..ef7aa358d0 100644
--- a/tests/e2e/ft/conftest_ft/app.py
+++ b/tests/e2e/ft/conftest_ft/app.py
@@ -1,6 +1,7 @@
 # NOTE: You MUST read tests/e2e/ft/README.md as source-of-truth and documentations
 
 import os
+import shutil
 from collections.abc import Callable
 from pathlib import Path
 from typing import Annotated
@@ -44,23 +45,26 @@ def run_pipeline(
 
     prepare(ft_mode)
 
-    for phase in effective_phases:
-        baseline_dump = f"{dump_dir}/{_dump_subdir('baseline', phase)}"
-        run_training(
-            train_args=build_baseline_args(ft_mode, baseline_dump, enable_dumper),
-            mode=ft_mode,
-            dump_dir=baseline_dump,
-        )
-
-        target_dump = f"{dump_dir}/{_dump_subdir('target', phase)}"
-        run_training(
-            train_args=build_target_args(ft_mode, target_dump, enable_dumper),
-            mode=ft_mode,
-            dump_dir=target_dump,
-        )
-
-    if enable_dumper:
-        compare_fn(dump_dir, ft_mode)
+    try:
+        for phase in effective_phases:
+            baseline_dump = f"{dump_dir}/{_dump_subdir('baseline', phase)}"
+            run_training(
+                train_args=build_baseline_args(ft_mode, baseline_dump, enable_dumper),
+                mode=ft_mode,
+                dump_dir=baseline_dump,
+            )
+
+            target_dump = f"{dump_dir}/{_dump_subdir('target', phase)}"
+            run_training(
+                train_args=build_target_args(ft_mode, target_dump, enable_dumper),
+                mode=ft_mode,
+                dump_dir=target_dump,
+            )
+
+        if enable_dumper:
+            compare_fn(dump_dir, ft_mode)
+    finally:
+        shutil.rmtree(dump_dir, ignore_errors=True)
 
 
 def create_comparison_app_and_run_ci(
diff --git a/tests/e2e/ft/conftest_ft/execution.py b/tests/e2e/ft/conftest_ft/execution.py
index c8fdb6df69..074391e278 100644
--- a/tests/e2e/ft/conftest_ft/execution.py
+++ b/tests/e2e/ft/conftest_ft/execution.py
@@ -104,7 +104,6 @@ def get_common_train_args(
             "--rollout-temperature 0.8 "
             "--rollout-batch-size 32 "
             "--n-samples-per-prompt 8 "
-            "--sglang-disable-cuda-graph "
             # Required for reproducibility (ref: https://github.com/THUDM/slime/pull/370)
             "--sglang-enable-deterministic-inference "
             "--sglang-attention-backend flashinfer "
@@ -174,6 +173,17 @@ def get_ft_args(mode: FTTestMode) -> str:
 }
 
 
+def get_train_env_vars_arg(mode: FTTestMode, *, deterministic: bool) -> str:
+    env_vars: dict[str, str] = {}
+    if deterministic:
+        env_vars.update(_DETERMINISTIC_ENV_VARS)
+    if mode.has_real_rollout:
+        env_vars["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
+    if not env_vars:
+        return ""
+    return f"--train-env-vars '{json.dumps(env_vars)}' "
+
+
 def run_training(
     train_args: str,
     mode: FTTestMode,
@@ -203,7 +213,7 @@ def run_training(
     }
     U.execute_train(
         train_args=train_args,
-        num_gpus_per_node=mode.train_gpus_per_node,
+        num_gpus_per_node=mode.total_node_gpus,
         megatron_model_type=mode.megatron_model_type,
         extra_env_vars=merged_env_vars,
         megatron_path=_MEGATRON_PATH,
diff --git a/tests/e2e/ft/conftest_ft/modes.py b/tests/e2e/ft/conftest_ft/modes.py
index 34baf9e761..07d7e0c2e8 100644
--- a/tests/e2e/ft/conftest_ft/modes.py
+++ b/tests/e2e/ft/conftest_ft/modes.py
@@ -40,6 +40,10 @@ def has_real_rollout(self) -> bool:
     def total_rollout_gpus(self) -> int:
         return self.rollout_num_engines * self.rollout_gpus_per_engine
 
+    @property
+    def total_node_gpus(self) -> int:
+        return self.train_gpus_per_node + self.total_rollout_gpus
+
 
 MODES: dict[str, FTTestMode] = {
     # --- 1-node (8 GPUs) variants ---
diff --git a/tests/e2e/ft/conftest_ft/scenario_deterministic.py b/tests/e2e/ft/conftest_ft/scenario_deterministic.py
index d63f80ade1..7a785b4886 100644
--- a/tests/e2e/ft/conftest_ft/scenario_deterministic.py
+++ b/tests/e2e/ft/conftest_ft/scenario_deterministic.py
@@ -5,7 +5,7 @@
 from pathlib import Path
 
 from tests.e2e.ft.conftest_ft.app import create_comparison_app_and_run_ci
-from tests.e2e.ft.conftest_ft.execution import get_common_train_args, get_ft_args
+from tests.e2e.ft.conftest_ft.execution import get_common_train_args, get_ft_args, get_train_env_vars_arg
 from tests.e2e.ft.conftest_ft.modes import FTTestMode
 
 from miles.utils.test_utils.comparisons.dumps import (
@@ -22,12 +22,6 @@
 TOTAL_NUM_ROLLOUTS: int = 2 * NUM_ROLLOUTS_PER_PHASE
 PHASE_START_ROLLOUT_IDS: dict[str, int] = {"phase_a": 0, "phase_b": NUM_ROLLOUTS_PER_PHASE}
 
-_DETERMINISTIC_ENV_VARS: str = (
-    '--train-env-vars \'{"NCCL_ALGO": "Ring", '
-    '"NVTE_ALLOW_NONDETERMINISTIC_ALGO": "0", '
-    '"CUBLAS_WORKSPACE_CONFIG": ":4096:8"}\' '
-)
-
 
 def _build_actions(phase_start_rollout_id: int) -> list[dict]:
     heal_trigger_rollout_id: int = phase_start_rollout_id + 1
@@ -64,7 +58,7 @@ def _build_phase_args(mode: FTTestMode, dump_dir: str, *, is_target: bool, enabl
     phase_start_rollout_id: int = PHASE_START_ROLLOUT_IDS[phase_name]
 
     base = get_common_train_args(mode, dump_dir=dump_dir, num_steps=TOTAL_NUM_ROLLOUTS, enable_dumper=enable_dumper)
-    base += "--deterministic-mode " + _DETERMINISTIC_ENV_VARS
+    base += "--deterministic-mode " + get_train_env_vars_arg(mode, deterministic=True)
     base += "--debug-deterministic-collective "
 
     if is_target:
diff --git a/tests/e2e/ft/conftest_ft/scenario_no_failure.py b/tests/e2e/ft/conftest_ft/scenario_no_failure.py
index 3e0c86163b..7ee2c022f6 100644
--- a/tests/e2e/ft/conftest_ft/scenario_no_failure.py
+++ b/tests/e2e/ft/conftest_ft/scenario_no_failure.py
@@ -3,7 +3,7 @@
 
 
 from tests.e2e.ft.conftest_ft.app import create_comparison_app_and_run_ci
-from tests.e2e.ft.conftest_ft.execution import get_common_train_args, get_ft_args
+from tests.e2e.ft.conftest_ft.execution import get_common_train_args, get_ft_args, get_train_env_vars_arg
 from tests.e2e.ft.conftest_ft.modes import FTTestMode
 
 from miles.utils.test_utils.comparisons.dumps import (
@@ -17,13 +17,17 @@
 
 
 def _build_baseline_args(mode: FTTestMode, dump_dir: str, enable_dumper: bool = True) -> str:
-    return get_common_train_args(mode, dump_dir=dump_dir, num_steps=NUM_STEPS, enable_dumper=enable_dumper)
+    return get_common_train_args(
+        mode, dump_dir=dump_dir, num_steps=NUM_STEPS, enable_dumper=enable_dumper
+    ) + get_train_env_vars_arg(mode, deterministic=False)
 
 
 def _build_target_args(mode: FTTestMode, dump_dir: str, enable_dumper: bool = True) -> str:
-    return get_common_train_args(
-        mode, dump_dir=dump_dir, num_steps=NUM_STEPS, enable_dumper=enable_dumper
-    ) + get_ft_args(mode)
+    return (
+        get_common_train_args(mode, dump_dir=dump_dir, num_steps=NUM_STEPS, enable_dumper=enable_dumper)
+        + get_ft_args(mode)
+        + get_train_env_vars_arg(mode, deterministic=False)
+    )
 
 
 def _compare(dump_dir: str, mode: FTTestMode) -> None:
diff --git a/tests/e2e/ft/conftest_ft/scenario_with_failure.py b/tests/e2e/ft/conftest_ft/scenario_with_failure.py
index cc1214cc9d..c48cf620b3 100644
--- a/tests/e2e/ft/conftest_ft/scenario_with_failure.py
+++ b/tests/e2e/ft/conftest_ft/scenario_with_failure.py
@@ -5,7 +5,7 @@
 from pathlib import Path
 
 from tests.e2e.ft.conftest_ft.app import create_comparison_app_and_run_ci
-from tests.e2e.ft.conftest_ft.execution import get_common_train_args, get_ft_args
+from tests.e2e.ft.conftest_ft.execution import get_common_train_args, get_ft_args, get_train_env_vars_arg
 from tests.e2e.ft.conftest_ft.modes import FTTestMode
 
 from miles.utils.test_utils.comparisons.dumps import (
@@ -88,6 +88,7 @@ def _expected_reconfigures(*, is_target: bool, phase: str, num_cells: int) -> li
 def _build_phase_args(mode: FTTestMode, dump_dir: str, *, is_target: bool, enable_dumper: bool = True) -> str:
     is_phase_a: bool = dump_dir.endswith("phase_a")
     base = get_common_train_args(mode, dump_dir=dump_dir, num_steps=NUM_PHASE_B_STEPS, enable_dumper=enable_dumper)
+    base += get_train_env_vars_arg(mode, deterministic=False)
 
     if is_target:
         base += get_ft_args(mode)
diff --git a/tests/e2e/ft/test_trainer_ft_deterministic_dp2_cp2_pp2.py b/tests/e2e/ft/test_trainer_ft_deterministic_dp2_cp2_pp2.py
index 198eac6daa..ae112d07bb 100644
--- a/tests/e2e/ft/test_trainer_ft_deterministic_dp2_cp2_pp2.py
+++ b/tests/e2e/ft/test_trainer_ft_deterministic_dp2_cp2_pp2.py
@@ -7,10 +7,9 @@
 from tests.e2e.ft.conftest_ft.scenario_deterministic import run_ci
 
 register_cuda_ci(
-    est_time=900,
-    suite="stage-c-8-gpu-h100",
+    est_time=2900,
+    suite="stage-c-8-gpu-h200",
     labels=["ft-short"],
-    disabled="FT e2e suite disabled in CI pending stabilization on current image (follow-up PR)",
 )
 
 _MODE: str = "dp2_cp2_pp2"
diff --git a/tests/e2e/ft/test_trainer_ft_deterministic_dp2_cp2_real_rollout.py b/tests/e2e/ft/test_trainer_ft_deterministic_dp2_cp2_real_rollout.py
index ac8bf1c520..ff29233736 100644
--- a/tests/e2e/ft/test_trainer_ft_deterministic_dp2_cp2_real_rollout.py
+++ b/tests/e2e/ft/test_trainer_ft_deterministic_dp2_cp2_real_rollout.py
@@ -7,10 +7,9 @@
 from tests.e2e.ft.conftest_ft.scenario_deterministic import run_ci
 
 register_cuda_ci(
-    est_time=1500,
-    suite="stage-c-8-gpu-h100",
+    est_time=3200,
+    suite="stage-c-8-gpu-h200",
     labels=["ft-short"],
-    disabled="real_rollout deterministic-inference sglang hang on current image",
 )
 
 _MODE: str = "dp2_cp2_real_rollout"
diff --git a/tests/e2e/ft/test_trainer_ft_deterministic_dp2_cp2_tp2_ep2.py b/tests/e2e/ft/test_trainer_ft_deterministic_dp2_cp2_tp2_ep2.py
index e86fc26025..f1bd26fe23 100644
--- a/tests/e2e/ft/test_trainer_ft_deterministic_dp2_cp2_tp2_ep2.py
+++ b/tests/e2e/ft/test_trainer_ft_deterministic_dp2_cp2_tp2_ep2.py
@@ -7,10 +7,9 @@
 from tests.e2e.ft.conftest_ft.scenario_deterministic import run_ci
 
 register_cuda_ci(
-    est_time=900,
-    suite="stage-c-8-gpu-h100",
+    est_time=2300,
+    suite="stage-c-8-gpu-h200",
     labels=["ft-short"],
-    disabled="FT e2e suite disabled in CI pending stabilization on current image (follow-up PR)",
 )
 
 _MODE: str = "dp2_cp2_tp2_ep2"
diff --git a/tests/e2e/ft/test_trainer_ft_deterministic_dp4_cp2.py b/tests/e2e/ft/test_trainer_ft_deterministic_dp4_cp2.py
index 339b83a9c7..58e9917364 100644
--- a/tests/e2e/ft/test_trainer_ft_deterministic_dp4_cp2.py
+++ b/tests/e2e/ft/test_trainer_ft_deterministic_dp4_cp2.py
@@ -7,10 +7,9 @@
 from tests.e2e.ft.conftest_ft.scenario_deterministic import run_ci
 
 register_cuda_ci(
-    est_time=900,
-    suite="stage-c-8-gpu-h100",
+    est_time=2900,
+    suite="stage-c-8-gpu-h200",
     labels=["ft-short"],
-    disabled="FT e2e suite disabled in CI pending stabilization on current image (follow-up PR)",
 )
 
 _MODE: str = "dp4_cp2"
diff --git a/tests/e2e/ft/test_trainer_ft_no_failure_dp2_cp2_pp2.py b/tests/e2e/ft/test_trainer_ft_no_failure_dp2_cp2_pp2.py
index 9d3ccfcdb1..5e091b2416 100644
--- a/tests/e2e/ft/test_trainer_ft_no_failure_dp2_cp2_pp2.py
+++ b/tests/e2e/ft/test_trainer_ft_no_failure_dp2_cp2_pp2.py
@@ -7,10 +7,9 @@
 from tests.e2e.ft.conftest_ft.scenario_no_failure import run_ci
 
 register_cuda_ci(
-    est_time=600,
-    suite="stage-c-8-gpu-h100",
+    est_time=1000,
+    suite="stage-c-8-gpu-h200",
     labels=["ft-short"],
-    disabled="FT e2e suite disabled in CI pending stabilization on current image (follow-up PR)",
 )
 
 _MODE: str = "dp2_cp2_pp2"
diff --git a/tests/e2e/ft/test_trainer_ft_no_failure_dp2_cp2_real_rollout.py b/tests/e2e/ft/test_trainer_ft_no_failure_dp2_cp2_real_rollout.py
index bf9787ed64..3b8aefa76e 100644
--- a/tests/e2e/ft/test_trainer_ft_no_failure_dp2_cp2_real_rollout.py
+++ b/tests/e2e/ft/test_trainer_ft_no_failure_dp2_cp2_real_rollout.py
@@ -8,9 +8,8 @@
 
 register_cuda_ci(
     est_time=1200,
-    suite="stage-c-8-gpu-h100",
+    suite="stage-c-8-gpu-h200",
     labels=["ft-short"],
-    disabled="real_rollout deterministic-inference sglang hang on current image",
 )
 
 _MODE: str = "dp2_cp2_real_rollout"
diff --git a/tests/e2e/ft/test_trainer_ft_no_failure_dp2_cp2_tp2_ep2.py b/tests/e2e/ft/test_trainer_ft_no_failure_dp2_cp2_tp2_ep2.py
index 30a9b78f05..5dbe5dfcaf 100644
--- a/tests/e2e/ft/test_trainer_ft_no_failure_dp2_cp2_tp2_ep2.py
+++ b/tests/e2e/ft/test_trainer_ft_no_failure_dp2_cp2_tp2_ep2.py
@@ -7,10 +7,9 @@
 from tests.e2e.ft.conftest_ft.scenario_no_failure import run_ci
 
 register_cuda_ci(
-    est_time=600,
-    suite="stage-c-8-gpu-h100",
+    est_time=800,
+    suite="stage-c-8-gpu-h200",
     labels=["ft-short"],
-    disabled="FT e2e suite disabled in CI pending stabilization on current image (follow-up PR)",
 )
 
 _MODE: str = "dp2_cp2_tp2_ep2"
diff --git a/tests/e2e/ft/test_trainer_ft_no_failure_dp4_cp2.py b/tests/e2e/ft/test_trainer_ft_no_failure_dp4_cp2.py
index 16222bf9d9..66bae2d7e8 100644
--- a/tests/e2e/ft/test_trainer_ft_no_failure_dp4_cp2.py
+++ b/tests/e2e/ft/test_trainer_ft_no_failure_dp4_cp2.py
@@ -7,10 +7,9 @@
 from tests.e2e.ft.conftest_ft.scenario_no_failure import run_ci
 
 register_cuda_ci(
-    est_time=600,
-    suite="stage-c-8-gpu-h100",
+    est_time=1100,
+    suite="stage-c-8-gpu-h200",
     labels=["ft-short"],
-    disabled="FT e2e suite disabled in CI pending stabilization on current image (follow-up PR)",
 )
 
 _MODE: str = "dp4_cp2"
diff --git a/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_real_rollout.py b/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_real_rollout.py
index 503a80dbb4..0a307d724c 100644
--- a/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_real_rollout.py
+++ b/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_real_rollout.py
@@ -7,7 +7,7 @@
 from tests.e2e.ft.conftest_ft.scenario_ft_random import run_ci
 
 register_cuda_ci(
-    est_time=2400, suite="stage-c-8-gpu-h100", labels=["ft-long"], disabled="FT soak tests pending CI infra support"
+    est_time=2400, suite="stage-c-8-gpu-h200", labels=["ft-long"], disabled="FT soak tests pending CI infra support"
 )
 
 _MODE: str = "dp2_cp2_real_rollout"
diff --git a/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_tp2_ep2.py b/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_tp2_ep2.py
index 735a91f972..2a9d0e7827 100644
--- a/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_tp2_ep2.py
+++ b/tests/e2e/ft/test_trainer_ft_random_dp2_cp2_tp2_ep2.py
@@ -7,7 +7,7 @@
 from tests.e2e.ft.conftest_ft.scenario_ft_random import run_ci
 
 register_cuda_ci(
-    est_time=1800, suite="stage-c-8-gpu-h100", labels=["ft-long"], disabled="FT soak tests pending CI infra support"
+    est_time=1800, suite="stage-c-8-gpu-h200", labels=["ft-long"], disabled="FT soak tests pending CI infra support"
 )
 
 _MODE: str = "dp2_cp2_tp2_ep2"
diff --git a/tests/e2e/ft/test_trainer_ft_realistic_gsm8k.py b/tests/e2e/ft/test_trainer_ft_realistic_gsm8k.py
index 1428f0a047..3149b53d0e 100644
--- a/tests/e2e/ft/test_trainer_ft_realistic_gsm8k.py
+++ b/tests/e2e/ft/test_trainer_ft_realistic_gsm8k.py
@@ -8,7 +8,7 @@
 
 register_cuda_ci(
     est_time=9000,
-    suite="stage-c-8-gpu-h100",
+    suite="stage-c-8-gpu-h200",
     labels=["ft-long", "long"],
     disabled="FT soak tests pending CI infra support",
 )
diff --git a/tests/e2e/ft/test_trainer_ft_with_failure_dp2_cp2_pp2.py b/tests/e2e/ft/test_trainer_ft_with_failure_dp2_cp2_pp2.py
index f3a0789aa6..4386bb6f13 100644
--- a/tests/e2e/ft/test_trainer_ft_with_failure_dp2_cp2_pp2.py
+++ b/tests/e2e/ft/test_trainer_ft_with_failure_dp2_cp2_pp2.py
@@ -7,10 +7,9 @@
 from tests.e2e.ft.conftest_ft.scenario_with_failure import run_ci
 
 register_cuda_ci(
-    est_time=900,
-    suite="stage-c-8-gpu-h100",
+    est_time=2100,
+    suite="stage-c-8-gpu-h200",
     labels=["ft-short"],
-    disabled="FT e2e suite disabled in CI pending stabilization on current image (follow-up PR)",
 )
 
 _MODE: str = "dp2_cp2_pp2"
diff --git a/tests/e2e/ft/test_trainer_ft_with_failure_dp2_cp2_real_rollout_dense.py b/tests/e2e/ft/test_trainer_ft_with_failure_dp2_cp2_real_rollout_dense.py
index 68dc1fa2c6..2c8102d05d 100644
--- a/tests/e2e/ft/test_trainer_ft_with_failure_dp2_cp2_real_rollout_dense.py
+++ b/tests/e2e/ft/test_trainer_ft_with_failure_dp2_cp2_real_rollout_dense.py
@@ -7,10 +7,9 @@
 from tests.e2e.ft.conftest_ft.scenario_with_failure import run_ci
 
 register_cuda_ci(
-    est_time=1500,
-    suite="stage-c-8-gpu-h100",
+    est_time=1700,
+    suite="stage-c-8-gpu-h200",
     labels=["ft-short"],
-    disabled="real_rollout deterministic-inference sglang hang on current image",
 )
 
 _MODE: str = "dp2_cp2_real_rollout_dense"
diff --git a/tests/e2e/ft/test_trainer_ft_with_failure_dp2_cp2_tp2_ep2.py b/tests/e2e/ft/test_trainer_ft_with_failure_dp2_cp2_tp2_ep2.py
index b1f4dc4b27..99352ffaa3 100644
--- a/tests/e2e/ft/test_trainer_ft_with_failure_dp2_cp2_tp2_ep2.py
+++ b/tests/e2e/ft/test_trainer_ft_with_failure_dp2_cp2_tp2_ep2.py
@@ -7,10 +7,9 @@
 from tests.e2e.ft.conftest_ft.scenario_with_failure import run_ci
 
 register_cuda_ci(
-    est_time=900,
-    suite="stage-c-8-gpu-h100",
+    est_time=1700,
+    suite="stage-c-8-gpu-h200",
     labels=["ft-short"],
-    disabled="FT e2e suite disabled in CI pending stabilization on current image (follow-up PR)",
 )
 
 _MODE: str = "dp2_cp2_tp2_ep2"
diff --git a/tests/fast/ray/train/test_group.py b/tests/fast/ray/train/test_group.py
index f03a8d3243..3c80eff00f 100644
--- a/tests/fast/ray/train/test_group.py
+++ b/tests/fast/ray/train/test_group.py
@@ -984,12 +984,16 @@ def test_with_normal_and_error_cells(self):
 
 
 def _checksum_response(engine_checksums: list[dict[str, str]]) -> list:
-    """Build a nested servers->groups->engines check_weights('checksum') response."""
+    """Build a nested groups->engines check_weights('checksum') response."""
     engines = [
-        {"success": True, "message": "ok", "ranks": [{"checksums": cs, "parallelism_info": {"rank": 0}}]}
+        {
+            "success": True,
+            "message": "ok",
+            "ranks": [{"checksums": cs, "parallelism_info": [{"role": "target", "rank": 0}]}],
+        }
         for cs in engine_checksums
     ]
-    return [[engines]]
+    return [engines]
 
 
 class TestMaybeLogInferenceEngineWeightChecksums:
diff --git a/tests/fast/utils/test_checksum_utils.py b/tests/fast/utils/test_checksum_utils.py
index 46d047c843..69f1e31931 100644
--- a/tests/fast/utils/test_checksum_utils.py
+++ b/tests/fast/utils/test_checksum_utils.py
@@ -15,53 +15,72 @@ def _engine_body(*, success: bool, ranks: list[dict[str, Any]] | None) -> dict[s
 
 
 def _rank(rank: int, checksums: dict[str, str]) -> dict[str, Any]:
-    return {"checksums": checksums, "parallelism_info": {"rank": rank}}
+    return {"checksums": checksums, "parallelism_info": [{"role": "target", "rank": rank}]}
 
 
 class TestFlattenInferenceEngineChecksums:
-    def test_single_server_group_engine_single_rank(self) -> None:
-        """One server/group/engine with one rank yields one prefixed checksum dict."""
-        result = [[[_engine_body(success=True, ranks=[_rank(0, {"w": "aaa"})])]]]
+    def test_single_group_engine_single_rank(self) -> None:
+        """One server-group/engine with one rank yields one prefixed checksum dict."""
+        result = [[_engine_body(success=True, ranks=[_rank(0, {"w": "aaa"})])]]
         assert flatten_inference_engine_checksums(result) == [{"rank0/w": "aaa"}]
 
     def test_multiple_engines_flattened_in_order(self) -> None:
-        """Engines across servers/groups are flattened into a single ordered list."""
+        """Engines across server-groups are flattened into a single ordered list."""
         result = [
-            [[_engine_body(success=True, ranks=[_rank(0, {"w": "e0"})])]],
-            [[_engine_body(success=True, ranks=[_rank(0, {"w": "e1"})])]],
+            [_engine_body(success=True, ranks=[_rank(0, {"w": "e0"})])],
+            [_engine_body(success=True, ranks=[_rank(0, {"w": "e1"})])],
         ]
         assert flatten_inference_engine_checksums(result) == [{"rank0/w": "e0"}, {"rank0/w": "e1"}]
 
     def test_none_node_rank_payloads_filtered(self) -> None:
         """None engine bodies (non-zero node ranks) are dropped before indexing."""
-        result = [[[_engine_body(success=True, ranks=[_rank(0, {"w": "e0"})]), None]]]
+        result = [[_engine_body(success=True, ranks=[_rank(0, {"w": "e0"})]), None]]
         assert flatten_inference_engine_checksums(result) == [{"rank0/w": "e0"}]
 
     def test_all_none_fails_loud(self) -> None:
         """A None-only result means the checksum action did nothing, so fail loud."""
-        result = [[[None, None]]]
+        result = [[None, None]]
         with pytest.raises(AssertionError, match="no non-None engine bodies"):
             flatten_inference_engine_checksums(result)
 
     def test_multi_rank_merged_with_rank_prefix(self) -> None:
         """Multiple ranks of one engine merge into one dict, prefixed by rank to avoid clobber."""
-        result = [[[_engine_body(success=True, ranks=[_rank(0, {"w": "r0"}), _rank(1, {"w": "r1"})])]]]
+        result = [[_engine_body(success=True, ranks=[_rank(0, {"w": "r0"}), _rank(1, {"w": "r1"})])]]
         assert flatten_inference_engine_checksums(result) == [{"rank0/w": "r0", "rank1/w": "r1"}]
 
     def test_ranks_out_of_order_sorted_by_parallelism_rank(self) -> None:
         """Ranks arriving out of order (zmq) are sorted by parallelism rank deterministically."""
-        out_of_order = [[[_engine_body(success=True, ranks=[_rank(1, {"w": "r1"}), _rank(0, {"w": "r0"})])]]]
-        in_order = [[[_engine_body(success=True, ranks=[_rank(0, {"w": "r0"}), _rank(1, {"w": "r1"})])]]]
+        out_of_order = [[_engine_body(success=True, ranks=[_rank(1, {"w": "r1"}), _rank(0, {"w": "r0"})])]]
+        in_order = [[_engine_body(success=True, ranks=[_rank(0, {"w": "r0"}), _rank(1, {"w": "r1"})])]]
         assert flatten_inference_engine_checksums(out_of_order) == flatten_inference_engine_checksums(in_order)
 
     def test_engine_failure_fails_loud(self) -> None:
         """An engine reporting success=False fails loud rather than silently dropping it."""
-        result = [[[_engine_body(success=False, ranks=[_rank(0, {"w": "aaa"})])]]]
+        result = [[_engine_body(success=False, ranks=[_rank(0, {"w": "aaa"})])]]
         with pytest.raises(AssertionError, match="reported failure"):
             flatten_inference_engine_checksums(result)
 
     def test_engine_without_ranks_fails_loud(self) -> None:
         """A success body with no ranks fails loud (nothing to compare)."""
-        result = [[[_engine_body(success=True, ranks=None)]]]
+        result = [[_engine_body(success=True, ranks=None)]]
         with pytest.raises(AssertionError, match="no ranks"):
             flatten_inference_engine_checksums(result)
+
+    def test_multi_role_parallelism_info_collapses_to_one_gpu_rank(self) -> None:
+        """Per-role parallelism_info (target + draft) sharing a GPU rank uses that single rank."""
+        rank_info = {
+            "checksums": {"w": "aaa"},
+            "parallelism_info": [{"role": "target", "rank": 0}, {"role": "draft", "rank": 0}],
+        }
+        result = [[_engine_body(success=True, ranks=[rank_info])]]
+        assert flatten_inference_engine_checksums(result) == [{"rank0/w": "aaa"}]
+
+    def test_parallelism_info_rank_disagreement_fails_loud(self) -> None:
+        """Roles reporting different GPU ranks for one engine rank is a contract violation."""
+        rank_info = {
+            "checksums": {"w": "aaa"},
+            "parallelism_info": [{"role": "target", "rank": 0}, {"role": "draft", "rank": 1}],
+        }
+        result = [[_engine_body(success=True, ranks=[rank_info])]]
+        with pytest.raises(AssertionError, match="one GPU rank"):
+            flatten_inference_engine_checksums(result)