
2026/10/20 18:08
DeepInfer:DeepSeek モデル向けの高性能推論フレームワーク
RSS: https://news.ycombinator.com/rss
要約▶
Japanese Translation:
DeepInfer は、NVIDIA SM100 GPU および Huawei Ascend 950 NPU 上でスケールサービング能力を備え、DeepSeek-V4.1-Flash モデルの DeepSeek の本番サービスを支えるオープンソースの高パフォーマンス推論フレームワークです。主に vLLM に構築され、SGLang および TensorRT-LLM を念頭に置いて設計されています。最適化されたカーネル(FlashMLA、DeepGEMM、DeepEP、TileKernels、DeepSelect、カスタム TileLang)と、Attention DP、バランスの取れた CP、MoE EP+EPLB、DSpark、信頼度に基づく検証、SWA 有界再生非同期スケジューリング、バッチ不変推論、決定論的サンプリング、構造化出力(regex/JSON スキーマ/EBNF)を含む高度な並列化戦略を支援します。
コア機能は環境変数経由で設定可能であり、DEEPINFER_ENABLE_MTP_VARLEN=1 で DSpark/検証を有効化し、DEEPINFER_USE_NEXTN_DECODE(デフォルト:5)で最大ドラフト長を制御します。デフォルトの V4.1 設定に対しては、DEEPINFER_YOCO_PREFILL_TRUNCATION_MODE=compact と DEEPINFER_YOCO_PREFILL_KEEP_TOKENS=140 でデコーダー SWA 有界再生非同期スケジューリングを設定します。デフォルトで CPU バッチスケジューリングとデバイス実行のオーバーラップを可能にするため、非同期スケジューリングを有効化し、NVIDIA 上で DEEPINFER_BATCH_INVARIANT=1 でバッチ構成間での同等の結果を実現するバッチ不変推論を活性化します。DEEPINFER_BATCH_INVARIANT=1、DEEPINFER_ENABLE_MTP_VARLEN=0、SamplingParams の固定 seq_seed を使用することで NVIDIA 上で決定論的サンプリングを実現し、SamplingParams に so_args={"schema_type": ..., "schema": ...} を渡すことで生成を regex/JSON スキーマ/EBNF に制限します。
完全な本番環境のセットアップには、専用の KV キャッシュシステムの統合、 disaggregated 形式での prefill/decode 操作の実施、およびデータ並列レプリカ間での負荷分散(初回リリースには含まれていない)が最終的には必要となる可能性があります。ユーザーは Linux、GCC ≥13.0、Python 3.12、CUDA 13(NVIDIA)または CANN 9.2(Ascend)、そして空の Python 3.12 環境の作成、DEEPINFER_ACCELERATOR=nvidia または ascend のエクスポート、および ./build.sh の実行からすぐに始めることができます。オフライン推論はデフォルトで deepseek-ai/DeepSeek-V4.1-Flash-1017 チェックポイントを使用し、初回実行時に Hugging Face キャッシュにダウンロードするか、DEEPINFER_HF_MODEL_DIR を介してオーバーライドします。例コマンドは python examples/offline_inference.py です。パフォーマンスベンチマークは 1 スケールアップドメイン内での prefill に 4 SM100 GPU、decode に 64 を使用して測定されました(再現は run_sweep.sh で可能;詳細は eval/README.md を参照)。このフレームワークは Apache License 2.0 の元でリリースされており、GitHub(https://github.com/deepseek-ai/DeepInfer)上で DeepSeek-AI(2026)に帰属しています。
本文
DeepInfer:DeepSeek 大規模推論フレームワークの紹介
概要
DeepInfer は、DeepSeek の大規模生産環境サービスを支える高パフォーマンスな推論フレームワークです。今回のオープンソースリリースには、スケール対応したサービス提供に不可欠の中核機能を含むオフライン推論機能が備わっています。
対応するモデル
- DeepSeek-V4.1-Flash
主要機能リスト
以下の特徴的な機能を搭載しています:
- 最適化されたカーネル・ランタイム: モデル、サンプリャー、およびカスタム TileLang カーネルへの対応。
- 高度なパラレル処理:
- アテンション DP(データ並列)/ バランスの取れた CP(テンソル並列)
- MoE EP + EPLB(專家並列+並列バランシング)
- DSpark と信頼度ベースのスケジューリングによる検証
- SWA 境界付きリプレイ: デコーダーのプリフィル作業量削減。
- アシンクロネバススケジュールリング: ゼロオーバーヘッドのバッチスケジューラを実装。
- バッチ非依存推論: モデル結果の構成不変性を保証。
- 決定論的サンプリング: 同一入力で常に同一出力を生成。
- 構造化された出力: JSON スキーマや正規表現による出力制限。
機能の詳細と設定方法
各機能に対する具体的な設定項目と環境変数を整理します。
1. 最適化されたカーネル
以下のコンポーネントが採用されています:
- FlashMLA、DeepGEMM、DeepEP、TileKernels、DeepSelect
- カスタム TileLang カーネル
2. DSpark と信頼度ベースの検証スケジューリング
ドラフトトークンによるデコード加速と、信頼度見積もりを用いた検証スケジュールを行います。
# 有効化設定 export DEEPINFER_ENABLE_MTP_VARLEN=1 # 最大ドラフト長の制御(デフォルト:5) export DEEPINFER_USE_NEXTN_DECODE=<設定値>
3. デコーダー用 SWA 境界付きリプレイ
V4.1 のデフォルト設定では、以下のパラメータを指定します。保持されるサフィックスは、SWA ウィンドウと最大バックトラック長の合計をカバーする必要があります。
export DEEPINFER_YOCO_PREFILL_TRUNCATION_MODE=compact export DEEPINFER_YOCO_PREFILL_KEEP_TOKENS=140
4. アシンクロネバススケジュールリング
CPU 側のバッチスケジューリングをデバイス実行とオーバーラップさせます。
- デフォルトで有効です。
5. バッチ非依存推論
NVIDIA GPU のみでの動作です。モデルの結果がバッチの構成によらず同一を保証します。
# NVIDIA GPU 限定の有効化設定 export DEEPINFER_BATCH_INVARIANT=1
6. 決定論的サンプリング
入力トークンおよびサンプリングパラメータが同一であれば、常に同一の出力を生成します。NVIDIA GPU のみでの動作です。
必須設定:
の有効化DEEPINFER_BATCH_INVARIANT=1
の設定DEEPINFER_ENABLE_MTP_VARLEN=0
実装例:
SamplingParams に固定された `seq_seed` を渡してください。
7. 構造化された出力
生成プロセスを正規表現、JSON スキーマ、または EBNF グラマーに制限します。
利用方法: SamplingParams に以下を渡して利用します(詳細な例はドキュメント参照)。
so_args={"schema_type": "...", "schema": "..."}
対応ハードウェア
| ハードウェア | チップセット |
|---|---|
| NVIDIA | SM100 GPU |
| Huawei Ascend | 950 NPU |
注意: 完全な推論サービスには以下のコンポーネントが必要ですが、本リリースでは含まれていません:
- KV キャッシュシステム
- VE/Prefill/Decode の分離(多数のエンジン機能は既に実装済み)
- プリフィルおよびデコードにおけるデータ並列レプリカ間での負荷分散
開始方法
前提条件
ビルド前に以下の環境を確保してください:
- OS: Linux
- コンパイラ: GCC >= 13.0
- 言語: Python 3.12
- CUDA: 13(NVIDIA GPU のみ)
- CANN: 9.2(Ascend プラットフォームのみのみ)
ソースコードからのビルド
空の Python 3.12 環境を作成し、アクティブ化してください。
# アセラータの指定(Ascend の場合は 'ascend' を使用) export DEEPINFER_ACCELERATOR=nvidia # ビルド実行 ./build.sh
オフライン推論
デフォルトのチェックポイント
deepseek-ai/DeepSeek-V4.1-Flash-1017 は、最初の実行時に Hugging Face キャッシュにダウンロードされます。
ローカルのコピーから動作させる場合は、以下の環境変数を設定してください:
export DEEPINFER_HF_MODEL_DIR=/path/to/DeepSeek-V4.1-Flash-1017
実行コマンド:
python examples/offline_inference.py
詳細は「例」のセクションをご覧ください。
パフォーマンスベンチマーク
測定環境:プリフィル用 4 つの SM100 GPU、デコード用 64 の SM100 GPU(それぞれ同一のスケーリング領域内)。
| メトリック | 値 |
|---|---|
| プリフィル | (ベンチマークデータ) |
| デコード | (ベンチマークデータ) |
- ベンチマークの再現については
を参照してください。run_sweep.sh - 評価ベンチマークの詳細は
をご参照ください。eval/README.md
ライセンス
DeepInfer は Apache License 2.0 に基づいてリリースされています。詳細は LICENSE ファイルをご確認ください。
謝辞
DeepInfer は主に以下のプロジェクトを基盤として構築され、多くのインスピレーションを得ています:
- vLLM
- SGLang
- TensorRT-LLM
これら各プロジェクトの開発者に感謝申し上げます。
引用
@misc{deepinfer2026, title={DeepInfer: High-performance, large-scale inference framework for DeepSeek models}, author={DeepSeek-AI}, year={2026}, publisher={GitHub}, howpublished={\url{https://github.com/deepseek-ai/DeepInfer}}, }