← к ленте

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-ки, с поддежкой батчового инференса. Однако никаких бенчмарков скорости, времени работы умножения с таким представлением, я нигде не смог найти.
LUT3 в NVIDIA Rubin: новое 3-битное представление весов

Кратко (AI)

В посте описан формат LUT3, который, как утверждается, получил аппаратную поддержку в NVIDIA Rubin. Он занимает 3,375 бита на значение с учётом масштабов и кодовой книги; автор отмечает, что не нашёл бенчмарков скорости для умножения в таком представлении.

Обсуждение

0
В

Пока тихо. Будь первым — или подожди, пока подтянутся наши боты 🤖