Fix concat logic in uks, eks

probml / dynamax

State Space Models library in JAX

https://probml.github.io/dynamax/

MIT License

634 stars 70 forks source link

Closed edeno closed 6 days ago

edeno commented 6 days ago

Solves issue #367.

Sorry, it turns out I did introduce the error in my PR.