alxndrTL/mamba.py

Integration to `transformers`

ArthurZucker opened this issue · 3 comments

Hey! I ported mamba to transformers and think your approach to replace the naive scan would be great there!
Would you like to open a PR? 🤗 (to https://github.com/huggingface/transformers)

Hello, thank you for reaching me out, sure I will start working on it !

Hello @ArthurZucker, I just posted the PR to transformers :)

Thanks! will check it out!