一个 Python 神经网络压缩框架,通过量化、剪枝和稀疏化加速 OpenVINO 推理,支持 PyTorch 与 ONNX 模型。
intel/neural-compressor
英特尔开源的 Python 模型压缩库,通过量化、稀疏化和蒸馏压缩深度学习模型。
- Star
- 2,722
- Fork
- 331
- 未关闭 Issue
- 30
- 最近提交
- 22小时前
这是什么?
Intel Neural Compressor 是一个用于压缩和加速深度学习模型的 Python 库。它提供静态量化、动态量化、SmoothQuant、仅权重量化、混合精度和 FP8 等方法,并支持 INT4、MXFP4、NVFP4 等低比特格式,主要面向大语言模型和视觉语言模型。
该库提供 PyTorch 和 JAX 的框架扩展,并带有类似 transformers 的接口,用于加载和量化模型。LLM 与 VLM 的量化通过集成 AutoRound 实现。文档列出的主要硬件包括 Intel Gaudi AI 加速器、Intel Core Ultra、Xeon Scalable 处理器和 Intel 数据中心 GPU,对 AMD CPU、ARM CPU 和 NVIDIA GPU 的测试较为有限。
项目还包含自动调优模块,可搜索满足精度目标的量化配置。文档涵盖 GPTQ、AWQ 风格的仅权重量化流程。
核心功能
- 01支持 PyTorch 和 JAX 模型的静态、动态和仅权重量化
- 02支持 SmoothQuant 与量化感知训练
- 03支持 FP8、MXFP8/MXFP4 和 NVFP4 等低精度格式(部分为实验性)
- 04提供类 transformers 接口,通过 AutoRound 量化 LLM 和 VLM
- 05自动调优,寻找满足精度目标的量化配置
适用场景
- 在 Intel 硬件上降低大语言模型推理的显存占用和延迟
- 部署前量化 PyTorch 或 JAX 视觉模型
- 在 Intel Gaudi 加速器上应用 FP8 量化
快速上手
- 1
从 PyPI 安装 PyTorch 后端包
pip install neural-compressor-pt - 2
从 PyPI 安装 JAX 后端包(可选)
pip install neural-compressor-jax - 3
运行 FP8 量化示例代码(需要 Intel Gaudi2 硬件或 Gaudi Docker 镜像)
使用前须知
- 入门示例主要面向 Intel Gaudi2 硬件,多数开发者可能没有该硬件。
- 部分功能仍处于实验阶段,例如 NVFP4、MXFP8/MXFP4 以及 Keras/JAX 的 FP8。
- 对 AMD CPU、ARM CPU 和 NVIDIA GPU 的支持仅经过有限测试。
- 特定硬件的环境配置(如匹配 Gaudi 软件版本)会增加安装工作量。
标签
常见问题
- neural-compressor 是什么?
- 英特尔开源的 Python 模型压缩库,通过量化、稀疏化和蒸馏压缩深度学习模型。 Intel Neural Compressor 是一个用于压缩和加速深度学习模型的 Python 库。它提供静态量化、动态量化、SmoothQuant、仅权重量化、混合精度和 FP8 等方法,并支持 INT4、MXFP4、NVFP4 等低比特格式,主要面向大语言模型和视觉语言模型。
- neural-compressor 可以免费使用吗?用的什么许可证?
- neural-compressor 以 Apache-2.0 许可证开源。用于商业产品前,请先确认许可证条款。
- neural-compressor 还在维护吗?
- 活跃维护中。neural-compressor 最近一次提交是在 2026年10月20日。
- neural-compressor 适合用来做什么?
- 在 Intel 硬件上降低大语言模型推理的显存占用和延迟 部署前量化 PyTorch 或 JAX 视觉模型 在 Intel Gaudi 加速器上应用 FP8 量化
- neural-compressor 怎么安装和运行?
- 从 PyPI 安装 PyTorch 后端包: pip install neural-compressor-pt 从 PyPI 安装 JAX 后端包(可选): pip install neural-compressor-jax 运行 FP8 量化示例代码(需要 Intel Gaudi2 硬件或 Gaudi Docker 镜像)
- 在哪里获取 neural-compressor?
- 源代码和安装说明在 GitHub 上:https://github.com/intel/neural-compressor
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
相关项目
Intel 开源的 LLM/VLM 训练后量化工具包,可将模型压缩为低比特权重,支持 Transformers、vLLM 和 SGLang。
一个极简、易于修改的 LLM 训练框架,覆盖分词、预训练、微调、评估到对话推理全流程,可在单台 GPU 节点上运行。
面向大模型分布式训练与推理的深度学习优化库,提升效率并降低使用门槛。
基于 PyTorch 的大模型分布式训练与推理系统,提供并行策略和显存优化等能力。
FlashAttention 官方实现,一种节省显存的精确注意力计算内核,用于 GPU 上 Transformer 模型的训练与推理。