ИИ Большие языковые модели
AXIOM
AXIOM — это загружаемое ядро Rust, которое оптимизирует вывод трансформеров, заменяя общие абстракции ОС примитивами, специфичными для вывода.
AXIOM
Что такое AXIOM?
AXIOM — это исследовательское ядро операционной системы, разработанное специально для эффективного выполнения вывода трансформеров на оборудовании с ограниченной памятью, использующее тензорно-нативное выделение памяти, планирование на границах слоёв и двухбуферную потоковую передачу весов.
AXIOM — сравнение с аналогами
| Модель ценообразования | Бесплатно | Бесплатно | Бесплатно | Бесплатно |
| Бесплатные кредиты | ||||
| Ключевые возможности |
|
|
|
|
| Плюсы |
|
|
|
|
| Минусы |
|
|
|
|
| Кому подходит |
|
|
|
|
Как использовать AXIOM?
- 1Настройте ночной инструментарий Rust и установите bootimage.
- 2Упакуйте веса модели с помощью предоставленного скрипта Python (pack_weights.py).
- 3Соберите ядро с помощью cargo +nightly run --release.
- 4Загрузите ядро в QEMU или на голое железо; оно выполнит вывод и выведет телеметрию.
AXIOM Ключевые возможности
- Тензорно-нативное выделение памяти с предварительно зарезервированными пулами
- Планирование на границах слоёв для предотвращения вытеснения внутри слоя
- Двухбуферная потоковая передача весов для наложения ввода-вывода и вычислений
- Планировщик LayerLock для выполнения с резидентным кэшем
- Квантованный вывод (Q4) для SmolLM2-135M и TinyLlama-1.1B
AXIOM Сценарии использования
- Запуск больших языковых моделей на системах с ограниченной памятью
- Оптимизация задержки вывода для моделей трансформеров
- Исследования в области оптимизации на уровне ОС для задач ИИ
- Оценка влияния планирования на уровне ядра на пропускную способность вывода
AXIOM Цены и бесплатный доступ
Модель оплаты AXIOM: Бесплатно.
AXIOM Плюсы и минусы
Плюсы
- Снижает накладные расходы на потоковую передачу с секунд до микросекунд на слой
- Оптимизирует расположение памяти для предсказуемых шаблонов доступа при выводе
- Открытый исходный код и расширяемость для исследований
- Демонстрирует значительное улучшение пропускной способности (в 14,8 раза TPS в бенчмарке)
Минусы
- В настоящее время ограничен конкретными моделями (SmolLM2-135M, TinyLlama-1.1B Q4)
- Требует NVMe на голом железе для предполагаемой оценки 7B-класса с низким потреблением памяти
- Ядра вычислений (проекция FFN) остаются узким местом
- Не является ОС общего назначения; отсутствуют пользовательское пространство, сеть, файловая система
Для чего лучше всего подходит AXIOM?
- Исследователи в области AI-систем и операционных систем
- Разработчики, оптимизирующие вывод на оборудовании с ограниченными ресурсами
- Инженеры, интересующиеся производительностью на уровне ядра для ИИ