GitHub repository: HGangloff/deep-dynamax
Description: Deep State Space Models library in JAX