Commit Graph

5 Commits

Author SHA1 Message Date
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