vvvm23/mamba-jax

Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX

语言：Python · GitHub 星标：93 · Fork：5

话题：jax · mamba

在 GitHub 上查看原仓库