NM

nvidia/model-optimizer

Developer tools

このページは AI により自動翻訳されています。技術的な詳細は原文をご確認ください。 原文を見る

5466 stars 品質 76 トレンド 57Apache-2.0

量子化、蒸留、剪定、ニューラルアーキテクチャ探索、投機的復号などの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

GitHub で README 全文を見る

推奨ツール

別のキーワードを試すか、フィルタを外してください。

インストール

npx skillfish add nvidia/model-optimizer
MCP 初心者?mcpskill.org のガイドから始めましょう →