Working on AMD ROCm Platform

alxndrTL / mamba.py

A simple and efficient Mamba implementation in pure PyTorch and MLX.

MIT License

960 stars 86 forks source link

Closed supersonictw closed 6 months ago

supersonictw commented 6 months ago

Thank you.

state-spaces/mamba is not working due to CUDA required. But this one you created is working successfully. Thank you!!! :smile:

Screenshot

alxndrTL commented 6 months ago

Hello, it's a pleasure to help!