dynamax
dynamax copied to clipboard
Add support for inference with padded arrays to support vmap over variable-length time series
This PR is still a WIP toward addressing issue https://github.com/probml/dynamax/issues/99