7 Commits

Author	SHA1	Message	Date
Nicolas Patry	a3c45da0a4	Improvments within mamba.	2024-01-31 10:28:58 +00:00
drbh	5b6f9259c1	feat: add optimization and first pass of integration test	2024-01-30 18:53:28 +00:00
drbh	966f3ba35c	feat: use fused kernel in forward pass	2024-01-29 21:54:23 +00:00
drbh	c2681b2bea	feat: build custom selective-scan kernels	2024-01-29 20:09:38 +00:00
drbh	1c32d53fc3	feat: prefer custom model and produce correct output	2024-01-25 17:07:37 -05:00
drbh	35939a28c7	feat: mvp single inference and explore integration	2024-01-24 20:55:12 -05:00
drbh	bcf145733c	feat: initial weight load	2024-01-23 01:37:09 +00:00