Intelが提供するオープンソースのPythonライブラリ。PyTorch、JAX、ONNXに対応し、量子化・スパース化・蒸留で深層学習モデルを圧縮します。
NVIDIA/Model-Optimizer
量子化、枝刈り、蒸留、スパース化、投機的デコードによって深層学習モデルを圧縮し、推論を高速化するNVIDIAのオープンソースライブラリ。
- スター
- 5,548
- フォーク
- 801
- 未解決の Issue
- 524
- 最終コミット
- 23 時間前
これは何?
NVIDIA Model Optimizer(ModelOpt)は、学習後量子化、量子化を考慮した学習、枝刈り、知識蒸留、Neural Architecture Search、スパース化、投機的デコードなどのモデル最適化手法を適用するPythonライブラリです。Hugging Face、PyTorch、ONNXのモデルを入力として受け取り、これらの手法を組み合わせて、最適化済みの量子化チェックポイントをエクスポートするPython APIを提供します。
エクスポートしたチェックポイントは、TensorRT-LLM、TensorRT、vLLM、SGLangなどの推論フレームワークでの利用を想定しています。また、学習時の最適化ではMegatron-LM、Megatron-Bridge、Hugging Face Accelerateと連携し、Hugging FaceのエクスポートAPIはtransformersモデルとdiffusersモデルの両方に対応しています。
リポジトリには、LLM、VLM、拡散モデル、ONNX向けのサンプルに加え、モデルタイプごとの対応状況をまとめたサポートマトリックスと、Hugging Faceで公開されている量子化済みチェックポイントが含まれています。
主な機能
- 01NVFP4やFP8形式を含む、学習後量子化と量子化を考慮した学習
- 02LLMを小型化するための構造化枝刈りと知識蒸留
- 03投機的デコードとスパース化の手法
- 04TensorRT-LLM、vLLM、SGLang向けに量子化済みチェックポイントをエクスポート
- 05Hugging Face、PyTorch、ONNXのモデル入力に対応
ユースケース
- 大規模なHugging Face LLMをNVFP4またはFP8に量子化してから、vLLMやTensorRT-LLMでサービングする
- 大きなモデルを枝刈り・蒸留し、品質をほぼ維持した小さなモデルにする
- ドラフトモジュールを追加し、投機的デコードでトークン生成を高速化する
- 画像・動画生成パイプラインで拡散モデルのレイテンシを削減する
クイックスタート
- 1
PyPIから安定版をインストールする
pip install -U nvidia-modelopt[all]
使う前に知っておきたいこと
- インストール時に、ライセンス条項を確認すべき追加のサードパーティ製オープンソースプロジェクトも導入されます。
- このソフトウェアを利用できるNVIDIAコンテナイメージには、それぞれ独自のライセンス条項があります。使用前に確認してください。
トピック
よくある質問
- Model-Optimizer とは?
- 量子化、枝刈り、蒸留、スパース化、投機的デコードによって深層学習モデルを圧縮し、推論を高速化するNVIDIAのオープンソースライブラリ。 NVIDIA Model Optimizer(ModelOpt)は、学習後量子化、量子化を考慮した学習、枝刈り、知識蒸留、Neural Architecture Search、スパース化、投機的デコードなどのモデル最適化手法を適用するPythonライブラリです。Hugging Face、PyTorch、ONNXのモデルを入力として受け取り、これらの手法を組み合わせて、最適化済みの量子化チェックポイントをエクスポートするPython APIを提供します。
- Model-Optimizer は無料で使えますか?ライセンスは?
- Model-Optimizer は Apache-2.0 ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。
- Model-Optimizer は今もメンテナンスされていますか?
- 活発にメンテナンス中。Model-Optimizer の最終コミットは 2026年11月6日 です。
- Model-Optimizer は何に使えますか?
- 大規模なHugging Face LLMをNVFP4またはFP8に量子化してから、vLLMやTensorRT-LLMでサービングする 大きなモデルを枝刈り・蒸留し、品質をほぼ維持した小さなモデルにする ドラフトモジュールを追加し、投機的デコードでトークン生成を高速化する 画像・動画生成パイプラインで拡散モデルのレイテンシを削減する
- Model-Optimizer のインストール方法は?
- PyPIから安定版をインストールする: pip install -U nvidia-modelopt[all]
- Model-Optimizer はどこで入手できますか?
- ソースコードとインストール手順は GitHub にあります:https://github.com/NVIDIA/Model-Optimizer
概要はプロジェクトの README をもとに AI が生成しています。GitHub の数値は毎日更新されます。
関連プロジェクト
SliceGPTは、重み行列の行と列をスライスしてLLMなどのTransformerモデルを縮小する、学習後圧縮手法です。
Triton Inference Serverのバックエンドとして、TensorRT-LLMやHuggingFace互換のLLMをTritonのHTTP・gRPC API経由で提供できます。
量子化や枝刈りでLLMを圧縮し、vLLMでそのまま推論に使えるチェックポイントを生成するPythonライブラリです。
長文脈LLM推論向けのKV cache圧縮、検索、マージ、量子化手法を比較する統合研究環境。
大規模言語モデルを量子化(GPTQ、AWQ、GGUF、FP8、EXL3など)し、NVIDIA、AMD、Intel、AppleのハードウェアやCPU上で推論するためのツールキット。