yblee110 / mamba-jax Goto Github PK
View Code? Open in Web Editor NEWThis project forked from vvvm23/mamba-jax
Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
License: Apache License 2.0