← к ленте

Open-source инференс движок для Gemma 4 26B-A4B на Swift/Metal

N@NeuralShitAI-инженер
1 мес

Разработчик представил open-source движок для запуска Gemma 4 26B-A4B на MacBook с M-процессорами, используя всего 2GB RAM.

Я сделал open-source инференс движок для Gemma 4 26B-A4B 4-bit. Движок использует всего 2gb ram при размере модели 14gb. Проект написан на Swift/Metal и запускается на любом MacBook с M-процессором, даже на M1 Air. Для того чтобы это работало, я использую стриминг весов с SSD и разные оптимизации. Кастомные fused kernels на Metal, репак весов в более подходящий формат, кеширование экспертов и I/O and compute overlap при работе с ними. Всего было более 100 разных экспериментов, все самое интересное описано в репе. Сейчас движок выдает 4-5 tok/s на M1 Air, 5-6 на M2 Air, 26-29 tok/s на M4 MacBook Pro и 31-35 tok/s на M5 MacBook Pro. Есть отдельное приложение на Mac, которое запускается буквально в одну команду. Надо только будет скачать веса с Hugging Face. Все абсолютно рабочее. На GitHub есть красиво оформленная страничка и больше 110 звездочек https://github.com/drumih/turbo-fieldfare

Кратко (AI)

Разработчик создал open-source движок на Swift и Metal для запуска модели Gemma 4 26B-A4B на устройствах Apple Silicon. Решение позволяет запускать 14-гигабайтную модель, используя лишь 2GB оперативной памяти, за счет стриминга весов с SSD и оптимизации вычислений.

Обсуждение

0
В

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