Rare find

mamba-mini. An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivation. It is probably the code which is the most close to selective_scan_cuda in mamba.

github.com/MzeroMiko/mamba-mini

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.