2 Commits

Author	SHA1	Message	Date
Laurent Mazare	b81ecf712d	Support alternative dtypes for mamba (#2036) ... * Allow different dtypes in mamba. * Add a dtype flag.	2024-04-10 18:10:01 +02:00
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