github.devotd.
NewInference & serving

nvidia/model-optimizer 0.47.0 — a model optimization library for inference

nvidia/model-optimizer is a Python library that builds checkpoints deployable to inference frameworks using a range of model optimization techniques.

Repositorynvidia/model-optimizer ↗

nvidia/model-optimizer is a Python library that builds checkpoints deployable to inference frameworks using a range of model optimization techniques.

GitHub description: A unified library of SOTA model optimization techniques like quantization, distillation, pruning, neural architecture search, speculative decoding, etc. It compresses deep learning models for downstream deployment frameworks like TensorRT-LLM, TensorRT, vLLM, etc. to optimize inference speed.

stars
★ 5.4K
Forks
759
Language
Python
License
Apache-2.0
Latest release
0.47.0
Read original source ↗View on GitHub ↗

nvidia/model-optimizer is a Python library that helps compress deep learning models for inference deployment. The latest release is 0.47.0, and the repository description states that it provides techniques such as quantization, pruning, distillation, Neural Architecture Search (NAS), and speculative decoding. The specific items that newly changed in this release are not in the provided material. The repository is published under the Apache-2.0 license.

This library takes Hugging Face, PyTorch, or ONNX models as input, composes optimization techniques through a Python API, and can export optimized quantized checkpoints. Quantization is a way of representing weights at lower precision, and the purpose given in the material is to reduce the weight memory that the checkpoint uses in the inference engine. The exported checkpoints can be deployed to inference frameworks such as SGLang, TensorRT-LLM, TensorRT, and vLLM. So the focus is on compressing and deploying models to fit the inference environment, rather than on training itself.

However, the provided material contains no measured results such as compression ratio, accuracy change, or throughput on specific models. The techniques and supported targets can be confirmed, but the size of the actual performance change cannot be judged from this information alone. The latest release number can be confirmed, but the changes in that version are not given either, so what has newly been added cannot be asserted.