
nvidia/model-optimizer
Developer toolsこのページは AI により自動翻訳されています。技術的な詳細は原文をご確認ください。 原文を見る
量子化、蒸留、剪定、ニューラルアーキテクチャ探索、投機的復号などのSOTAモデル最適化技術の統一ライブラリ。
概要
ドキュメント |ロードマップ |発表ブログ (または と呼ばれます)は、量子化、剪定、ニューラルアーキテクチャ探索(NAS)、蒸留、推測的復号、スパーシティなどの最先端のモデル最適化技術を含み、モデルを加速させるライブラリです。 モデルオプティマイザーは現在、Hugging Face、PyTorch、またはONNXモデルの入力をサポートしています。 Model Optimizerは、上記のモデル最適化手法を簡単に作成し、最適化された量子化されたチェックポイントをエクスポートできるPython APIを提供しています。Model OptimizerはNVIDIA Megatron-Bridge、Megatron-LM、Hugging Face Accelerateとも連携し、必要な推論最適化技術の訓練に利用できます。 NVIDIA AIソフトウェアエコシステムにシームレスに統合されたModel Optimizerで生成された量子化されたチェックポイントは、SGLang、TensorRT-LLM、TensorRT、vLLMなどの下流推論フレームワークへの展開が可能となっています。統合されたHugging FaceエクスポートAPIは、トランスフォーマーモデルとディフューザーモデルの両方をサポートしています。
概要(AI 翻訳)
ドキュメント | ロードマップ | 発表ブログ
NVIDIA Model Optimizer(通称:Model Optimizer、またはModelOpt)は、量子化、プルーニング、ニューラルアーキテクチャ探索(NAS)、蒸留、推測的復号、スパーシティなどの最先端のモデル最適化技術を含み、モデルを加速させるライブラリです。
[入力] モデルオプティマイザーは現在、Hugging Face、PyTorch、またはONNXモデルの入力をサポートしています。
[最適化] モデルオプティマイザーは、上記のモデル最適化手法を簡単に作成し、最適化された量子化されたチェックポイントをエクスポートできるPython APIを提供します。 モデルオプティマイザーはまた、NVIDIA Megatron-Bridge、Megatron-LM、Hugging Face Accelerateと連携し、必要な推論最適化技術の訓練に活用されています。
[デプロイ用エクスポート] NVIDIA AIソフトウェアエコシステムにシームレスに統合されたModel Optimizerで生成された量子化されたチェックポイントは、SGLang、TensorRT-LLM、TensorRT、vLLMなどの下流推論フレームワークへの展開が可能となっています。統合されたHugging FaceエクスポートAPIは、トランスフォーマーモデルとディフューザーモデルの両方をサポートしています。
最新ニュース
このページは AI により自動翻訳されています。技術的な詳細は原文をご確認ください。
README
以下は原文 README の抜粋です。原文を見る。
NVIDIA Model Optimizer (referred to as Model Optimizer, or ModelOpt) is a library comprising state-of-the-art model optimization techniques including quantization, pruning, Neural Architecture Search (NAS), distillation, speculative decoding and sparsity to accelerate models.
[Input] Model Optimizer currently supports inputs of a Hugging Face, PyTorch or ONNX model.
[Optimize] Model Optimizer provides Python APIs for users to easily compose the above model optimization techniques and export an optimized quantized checkpoint. Model Optimizer is also integrated with NVIDIA Megatron-Bridge, Megatron-LM and Hugging Face Accelerate for training required inference optimization techniques.
[Export for deployment] Seamlessly integrated within the NVIDIA AI software ecosystem, the quantized checkpoint generated from Model Optimizer is ready for deployment in downstream inference frameworks like SGLang, TensorRT-LLM, TensorRT, or vLLM. The unified Hugging Face export API now supports both transformers and diffusers models.
Latest News
- [2026/10/28] BLOG: GLM-5.3-Flash at 2-3 Bits per Expert Weight: IQ2_S PTQ Recipes and QAD: IQ2_S experts with NVFP4 down projections match the FP8 checkpoint without training, at less than half its size; QAD recovers long-context accuracy on the all-IQ2_S model.
推奨ツール
別のキーワードを試すか、フィルタを外してください。
インストール
npx skillfish add nvidia/model-optimizer