diff --git a/scripts/benchmarks/__init__.py b/scripts/benchmarks/__init__.py
index e4c53bbaf..7bb8fc1a2 100644
--- a/scripts/benchmarks/__init__.py
+++ b/scripts/benchmarks/__init__.py
@@ -1 +1,5 @@
 """Benchmark helpers."""
+
+from .deepseek_v3_moe import DTYPE_MAP, benchmark_deepseek_v3
+
+__all__ = ["benchmark_deepseek_v3", "DTYPE_MAP"]
diff --git a/scripts/benchmarks/deepseek_v3_moe_sweep.py b/scripts/benchmarks/deepseek_v3_moe_sweep.py
index 257400dcf..c6eac44ba 100644
--- a/scripts/benchmarks/deepseek_v3_moe_sweep.py
+++ b/scripts/benchmarks/deepseek_v3_moe_sweep.py
@@ -6,9 +6,15 @@ from __future__ import annotations
 
 import argparse
 import csv
+import sys
 from pathlib import Path
 from types import SimpleNamespace
 
+CURRENT_DIR = Path(__file__).resolve().parent
+PACKAGE_ROOT = CURRENT_DIR.parent
+if str(PACKAGE_ROOT) not in sys.path:
+    sys.path.insert(0, str(PACKAGE_ROOT))
+
 from scripts.benchmarks.deepseek_v3_moe import (
     DTYPE_MAP,
     benchmark_deepseek_v3,