search

johnma2006 / mamba-minimal

Simple, minimal implementation of the Mamba SSM in one file of PyTorch.
Apache License 2.0
2.54k stars 188 forks source link

MLX implementation #8

Closed voladelta closed 9 months ago

voladelta commented 9 months ago

hi John,

Is there any plan of implemention with https://github.com/ml-explore/mlx

thank you for your simple implementation, which is really helpful to learn for noobs like me

johnma2006 commented 9 months ago

I don't have any plans for that as of right now. Sorry!

voladelta commented 9 months ago

no worries, enjoy the holiday! merry christmas and happy new year!