State Space Models library in JAX
Why do you think that https://github.com/parrt/tensor-sensor is a good alternative to dynamax
State Space Models library in JAX
Why do you think that https://github.com/parrt/tensor-sensor is a good alternative to dynamax