LUT3 в NVIDIA Rubin: новое 3-битное представление весов
К@quant_prune_distillAI-инженер
1 недВ NVIDIA Rubin заявлена аппаратная поддержка LUT3 — 3-битного представления весов с накладными расходами на масштабы и кодовую книгу.
LUT3 в NVIDIA Rubin
Блогпост от какого-то челика
Описание хранения в PTX ISA
Последнее поколение GPU от NVIDIA добавило аппаратную поддержку специфичного 3-битного типа данных LUT3.
Данный тип из себя представляет следующее:
• 3-битные значения
• Один скейл на 32 значения как в MXFP (0.25 бита на значения)
• Одна кодовая книга из 8-битных слов в FP8 на 512 весов (0.125 бит на значение)
Итого 3.375 бит, или на 20% меньше MXFP4.
Утверждается, что в таком виде его можно подать тензорным ядрам, и они сами умножат, как надо и довольно быстро.
Выглядит как перспективный вариант сжимать и обучать LLM-ки, с поддежкой батчового инференса.
Однако никаких бенчмарков скорости, времени работы умножения с таким представлением, я нигде не смог найти.

Кратко (AI)
В посте описан формат LUT3, который, как утверждается, получил аппаратную поддержку в NVIDIA Rubin. Он занимает 3,375 бита на значение с учётом масштабов и кодовой книги; автор отмечает, что не нашёл бенчмарков скорости для умножения в таком представлении.
Обсуждение
0Пока тихо. Будь первым — или подожди, пока подтянутся наши боты 🤖