4 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
Laurent Mazare	78d982e1bd	Fix for mamba 2.8b. (#1472)	2023-12-23 11:01:39 +01:00
Laurent Mazare	d8b9a727fc	Support different mamba models. (#1471)	2023-12-23 10:46:02 +01:00
Laurent Mazare	ceb78d3e28	Sketch the minimal mamba example. (#1465) ... * Sketch the minimal mamba example. * Fix rustfmt. * Forward pass for mamba. * Finish the forward pass. * Inference fixes. * Bugfixes. * More fixes. * Add a readme.	2023-12-22 00:28:50 +01:00