Открытый код для эффективной генерации изображений и видео высокого разрешения с помощью линейных diffusion transformer, включая пайплайны обучения и инференса.
ridgerchu/matmulfreellm
Открытая реализация языковых моделей MatMul-free: вместо матричных умножений используются тернарные веса и сложение; совместима с Hugging Face Transformers.
- Звёзды
- 3,101
- Форки
- 204
- Открытые issues
- 25
- Последний коммит
- 2 месяца назад
Что это?
MatMul-Free LM — это архитектура языковой модели, в которой отсутствуют операции матричного умножения (MatMul), что позволяет снизить вычислительные затраты на большие языковые модели. В репозитории представлена реализация на PyTorch, интегрируемая с библиотекой Hugging Face Transformers: модели можно инициализировать через AutoModel и использовать со стандартными API генерации.
В проекте есть предобученные чекпойнты на 370M, 1.3B и 2.7B параметров; модели на 370M и 1.3B обучены соответственно на 15B и 100B токенов. Ядра написаны на Triton, а пакет зависит от PyTorch 2.0 или новее и einops. Код адаптирован из flash-linear-attention и соответствует статье Scalable MatMul-free Language Modeling (arXiv 2406.02528).
Ключевые возможности
- 01Архитектура языковой модели без операций матричного умножения
- 02Совместимость с API Hugging Face AutoModel и generate
- 03Предобученные модели на 370M, 1.3B и 2.7B параметров
- 04Ядра Triton на основе кода flash-linear-attention
- 05Установка напрямую из репозитория GitHub
Сценарии использования
- Исследование эффективных архитектур языковых моделей без MatMul
- Эксперименты с LLM на тернарных весах на GPU с использованием инструментов Transformers
- Предварительное обучение или дообучение моделей MatMul-free на собственных данных
- Изучение нейроморфных аппаратных платформ или платформ с ограниченными вычислительными ресурсами для LLM
Быстрый старт
- 1
Установите необходимые зависимости: PyTorch 2.0 или новее, Triton 2.2 или новее и einops
- 2
Установите библиотеку напрямую из GitHub
pip install -U git+https://github.com/ridgerchu/matmulfreellm
Что важно знать
- Для приведённого примера генерации нужна GPU с поддержкой CUDA, поскольку в нём вызываются .cuda() и .half().
- Проект зависит от Triton, который добавляет довольно объёмный набор зависимостей для GPU.
Темы
Частые вопросы
- Что такое matmulfreellm?
- Открытая реализация языковых моделей MatMul-free: вместо матричных умножений используются тернарные веса и сложение; совместима с Hugging Face Transformers. MatMul-Free LM — это архитектура языковой модели, в которой отсутствуют операции матричного умножения (MatMul), что позволяет снизить вычислительные затраты на большие языковые модели. В репозитории представлена реализация на PyTorch, интегрируемая с библиотекой Hugging Face Transformers: модели можно инициализировать через AutoModel и использовать со стандартными API генерации.
- matmulfreellm можно использовать бесплатно? Какая у него лицензия?
- matmulfreellm распространяется с открытым исходным кодом под лицензией Apache-2.0. Перед использованием в коммерческих продуктах проверьте условия лицензии.
- matmulfreellm всё ещё поддерживается?
- Обновлялся за последние 6 месяцев. Последний коммит в matmulfreellm — 6 сент. 2026 г..
- Для чего можно использовать matmulfreellm?
- Исследование эффективных архитектур языковых моделей без MatMul Эксперименты с LLM на тернарных весах на GPU с использованием инструментов Transformers Предварительное обучение или дообучение моделей MatMul-free на собственных данных Изучение нейроморфных аппаратных платформ или платформ с ограниченными вычислительными ресурсами для LLM
- Как установить matmulfreellm?
- Установите необходимые зависимости: PyTorch 2.0 или новее, Triton 2.2 или новее и einops Установите библиотеку напрямую из GitHub: pip install -U git+https://github.com/ridgerchu/matmulfreellm
- Где скачать matmulfreellm?
- Исходный код и инструкции по установке — на GitHub: https://github.com/ridgerchu/matmulfreellm
Описание создано ИИ на основе README проекта. Статистика GitHub обновляется ежедневно.
Похожие проекты
Компактный open-source проект по обучению LLM с нуля: на одной GPU примерно за 2 часа можно обучить языковую модель на 64 млн параметров.
Открытые веса языковых моделей gpt-oss-120b и gpt-oss-20b от OpenAI, а также эталонные реализации инференса, инструментов и клиентского кода.
Библиотека операторов на Triton с ядрами, совместимыми с PyTorch и LLM, для работы на разных AI-ускорителях.
Личный учебный и инженерный блокнот с CUDA-ядрами, написанными вручную, примерами CUTLASS/CuTe и Triton, а также заметками об оптимизации LLM.
Библиотека Hugging Face для Parameter-Efficient Fine-Tuning (PEFT): адаптация крупных предварительно обученных моделей с помощью обучения небольшого набора дополнительных параметров.