https://github.com/tommyip/mxmamba

Mamba implementation in MLX
https://github.com/tommyip/mxmamba

Last synced: about 2 months ago
JSON representation

Mamba implementation in MLX

• Host: GitHub
• URL: https://github.com/tommyip/mxmamba
• Owner: tommyip
• License: apache-2.0
• Created: 2024-02-04T17:01:01.000Z (11 months ago)
• Default Branch: master
• Last Pushed: 2024-04-01T15:26:01.000Z (9 months ago)
• Last Synced: 2024-04-01T16:40:05.878Z (9 months ago)
• Language: Python
• Homepage:
• Size: 24.4 KB
• Stars: 1
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# mxmamba

Mamba implementation optimized for Apple silicon.

> Mamba: Linear-Time Sequence Modeling with Selective State Spaces  

> Albert Gu*, Tri Dao*  

> Paper: https://arxiv.org/abs/2312.00752  

> Office repo: [state-spaces/mamba](https://github.com/state-spaces/mamba)