GitHub repository: xinglong-li/ssm-jax
Primary language: Jupyter Notebook
Description: State Space Models library in JAX