implementations of SSMs and related algorithms in JAX
models:
- mamba [jax]
- mamba2 [jax, equinox]
- xlstm [equinox]
-
Notifications
You must be signed in to change notification settings - Fork 0
okarthikb/state-space-models
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published