Giters

PeaBrane / mamba-tiny

Simple, minimal implementation of the Mamba SSM in one pytorch file. More efficient than using for loops, but probably less efficient than using associative scans

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PeaBrane/mamba-tiny Stargazers