GitHub repository: Joshuaalbert/essm_jax
Primary language: Python
Description: Extended State Space Model for JAX