mamba2-jax: A pure JAX/Flax implementation of Mamba-2 for language modeling and time series forecasting.
语言:Python · GitHub 星标:19 · Fork:3
在 GitHub 上查看原仓库