inisis/OnnxSlim
A Toolkit to Help Optimize Onnx Model
解决的问题
OnnxSlim 通过移除不必要的算子来降低 ONNX 模型的复杂度。此过程在保持相同准确性的同时精简模型,从而提升推理速度并加快模型加载时间。
工作原理
提供 Python 包和命令行界面来处理 ONNX 模型。用户可以运行简单的 bash 命令来精简模型,或在 Python 脚本中使用 onnxslim.slim() 函数来优化模型的图结构。
适用对象
专为需要优化部署模型的 AI 开发者和 ML 工程师设计,特别是使用 ONNX 格式以支持各种硬件加速器或推理引擎的用户。
亮点
- 集成到 NVIDIA TensorRT-Model-Optimizer、HuggingFace Optimum 和 transformers.js 等主要框架中。
- 在 LLM 推理优化挑战中拥有经过验证的性能(在 AICAS 2024 和 2025 中排名第一)。
- 被 Ultralytics 和 MNN-LLM 等项目广泛采用。
- 通过 pip 简单安装。
相关
- 项目
- 项目
- 项目
- 项目
- 项目