State Space Models library in JAX
Why do you think that https://github.com/AakashKumarNain/TF_JAX_tutorials is a good alternative to dynamax
State Space Models library in JAX
Why do you think that https://github.com/AakashKumarNain/TF_JAX_tutorials is a good alternative to dynamax