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