state-spaces

/

mamba2-130m

like 11

Follow

State Space Models 379

Hi! I wanted to share an Android + OpenCL implementation of Mamba2-130M, in case anyone wants to run it on a phone:

• Try it: Edgi on Google Play — runs fully on-device, no cloud.
• Open source: the app is built on top of the open-source adreno-llms inference engine — https://github.com/a8nova/adreno-llms — pure C++/OpenCL with hand-written kernels tuned for Adreno, including custom kernels for the SSD recurrence and gated RMSNorm.

It's tuned and tested on Adreno 6xx GPUs — the GPU class in mid-range and older Android phones (verified on a 2020 Motorola Razr / Adreno 620) — and should run on most arm64 Android phones with OpenCL, though the optimizations are Adreno-specific. On the Adreno 620 it decodes at ~24.3 tok/s in fp16 — among the fastest models in the collection.

Hope it's useful — happy to answer questions!