huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 02:38:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
2,124 Commits 74 Branches 16 Tags
metal-gemm-testing
Commit Graph

1 Commits

Author SHA1 Message Date
Laurent Mazare
1e26d539d9 Improved mamba model optimized for inference (#1694) 
* Sketch the mamba model for inference.

* Complete the forward pass.

* Add the mamba example.

* Optimize the selective-scan part.

* Fix a couple shape mismatches and get inference to work.

* Tweak the readmes.

* More readme tweaks.
 2024-02-11 17:04:57 +01:00