Add support for inference with padded arrays to support vmap over variable-length time series

probml / dynamax

State Space Models library in JAX

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

MIT License

634 stars 70 forks source link

Open slinderman opened 9 months ago

slinderman commented 9 months ago

This PR is still a WIP toward addressing issue https://github.com/probml/dynamax/issues/99