Python-фреймворк для сжатия нейросетей с помощью квантования, прореживания и разреживания, ускоряющий инференс в OpenVINO, PyTorch и ONNX.
intel/neural-compressor
Открытая Python-библиотека Intel для сжатия моделей глубокого обучения с помощью квантования, разреживания и дистилляции в PyTorch, JAX и ONNX.
- Звёзды
- 2,722
- Форки
- 331
- Открытые issues
- 30
- Последний коммит
- 23 часа назад
Что это?
Intel Neural Compressor — Python-библиотека для уменьшения размера и ускорения моделей глубокого обучения. Она предоставляет методы квантования после обучения и квантования с учётом обучения, включая статическое и динамическое квантование, SmoothQuant, квантование только весов, смешанную точность и FP8, а также низкобитные форматы INT4, MXFP4 и NVFP4 для больших языковых и визуально-языковых моделей.
В библиотеку входят расширения для PyTorch и JAX, а также API в стиле transformers для загрузки и квантования моделей. Квантирование LLM и VLM интегрировано с AutoRound. В документации указаны такие целевые аппаратные платформы, как ускорители Intel Gaudi AI, Intel Core Ultra, процессоры Xeon Scalable и GPU Intel Data Center; также в ограниченном объёме тестировались CPU AMD и ARM, а также GPU NVIDIA.
В проект также входит модуль автоматической настройки, который подбирает конфигурации квантования, соответствующие заданному порогу точности. В документации описаны сценарии квантования только весов в стиле GPTQ и AWQ.
Ключевые возможности
- 01Статическое, динамическое квантование и квантование только весов для моделей PyTorch и JAX
- 02Поддержка SmoothQuant и обучения с учётом квантования
- 03Низкоточные форматы FP8, MXFP8/MXFP4 и NVFP4 (некоторые из них экспериментальные)
- 04API в стиле Transformers для квантования LLM и VLM с помощью AutoRound
- 05Автоматический подбор параметров квантования, соответствующих заданному порогу точности
Сценарии использования
- Снижение потребления памяти и задержки при инференсе больших языковых моделей на оборудовании Intel
- Квантование моделей компьютерного зрения на PyTorch или JAX перед развёртыванием
- Применение квантования FP8 на ускорителях Intel Gaudi
Быстрый старт
- 1
Установите пакет бэкенда PyTorch из PyPI
pip install neural-compressor-pt - 2
Установите пакет бэкенда JAX из PyPI (необязательно)
pip install neural-compressor-jax - 3
Запустите пример кода квантования FP8 (требуется оборудование Intel Gaudi2 или образ Docker с Gaudi)
Что важно знать
- Основной пример для начала работы рассчитан на оборудование Intel Gaudi2, которого, скорее всего, нет у большинства разработчиков.
- Некоторые функции, такие как NVFP4, MXFP8/MXFP4 и FP8 для Keras/JAX, пока находятся на экспериментальной стадии.
- Поддержка CPU AMD, CPU ARM и GPU NVIDIA проверена лишь в ограниченном объёме.
- Настройка оборудования, например согласование версий ПО Gaudi, усложняет установку.
Темы
Частые вопросы
- Что такое neural-compressor?
- Открытая Python-библиотека Intel для сжатия моделей глубокого обучения с помощью квантования, разреживания и дистилляции в PyTorch, JAX и ONNX. Intel Neural Compressor — Python-библиотека для уменьшения размера и ускорения моделей глубокого обучения. Она предоставляет методы квантования после обучения и квантования с учётом обучения, включая статическое и динамическое квантование, SmoothQuant, квантование только весов, смешанную точность и FP8, а также низкобитные форматы INT4, MXFP4 и NVFP4 для больших языковых и визуально-языковых моделей.
- neural-compressor можно использовать бесплатно? Какая у него лицензия?
- neural-compressor распространяется с открытым исходным кодом под лицензией Apache-2.0. Перед использованием в коммерческих продуктах проверьте условия лицензии.
- neural-compressor всё ещё поддерживается?
- Активно поддерживается. Последний коммит в neural-compressor — 20 окт. 2026 г..
- Для чего можно использовать neural-compressor?
- Снижение потребления памяти и задержки при инференсе больших языковых моделей на оборудовании Intel Квантование моделей компьютерного зрения на PyTorch или JAX перед развёртыванием Применение квантования FP8 на ускорителях Intel Gaudi
- Как установить neural-compressor?
- Установите пакет бэкенда PyTorch из PyPI: pip install neural-compressor-pt Установите пакет бэкенда JAX из PyPI (необязательно): pip install neural-compressor-jax Запустите пример кода квантования FP8 (требуется оборудование Intel Gaudi2 или образ Docker с Gaudi)
- Где скачать neural-compressor?
- Исходный код и инструкции по установке — на GitHub: https://github.com/intel/neural-compressor
Описание создано ИИ на основе README проекта. Статистика GitHub обновляется ежедневно.
Похожие проекты
Набор инструментов Intel для посттренировочной квантизации LLM и VLM, который сжимает модели до низкой разрядности весов для Transformers, vLLM и SGLang.
Минималистичный, удобный для модификации инструментарий для полного цикла работы с LLM — от токенизации и предобучения до дообучения, оценки и чата; запускается на одном узле с GPU.
Библиотека для оптимизации глубокого обучения, упрощающая и ускоряющая распределённое обучение и инференс больших моделей.
Система на базе PyTorch для распределённого обучения и инференса крупных AI-моделей с поддержкой параллелизма, оптимизации памяти и гетерогенного обучения.
Официальная реализация FlashAttention — точного и экономного по памяти attention-ядра для обучения и инференса Transformer-моделей на GPU.