search

alxndrTL / mamba.py

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

A basic implementation of ONNX exportation. #34

Closed Zeng-WCh closed 3 months ago

Zeng-WCh commented 5 months ago
alxndrTL commented 3 months ago

Hello, thank you for the PR! (Sorry for the late response)