search

alxndrTL / mamba.py

A simple and efficient Mamba implementation in pure PyTorch and MLX.
MIT License
960 stars 86 forks source link

mlx support : pscan, mamba and mamba_lm files rewritten in MLX #4

Closed alxndrTL closed 9 months ago