sts-jax
sts-jax copied to clipboard
probml
Structural Time Series in JAX
sts_components like `SeasonalDummy` and `Autoregressive` are not part of the module `sts_jax.structural_time_series.sts_model`
In blackjax 1.0.0 there were changes in the API that broke the fit_hmc function, this PR fix it.
Hi, I am trying several components in [sts_components.py](https://github.com/probml/sts-jax/blob/main/sts_jax/structural_time_series/sts_components.py). They are all working, but `sts.Cycle()` is not. First, it complains that the Uniform distribution has no `.mode()` (see line 770 and...
Hi, I would like to grab developers' attention to the issue that both `"fit_vi"` and `"fit_hmc"` functions are broken. For such a reason, unfortunately the [CO_2 demo](https://github.com/probml/sts-jax/blob/main/sts_jax/structural_time_series/demos/sts_co2_demo.ipynb) is not working....
I installed sts-jax using the latest development branch. When running the sts_electric_demo notebook, I get an error message ImportError: cannot import name 'PRNGKey' from 'dynamax.types' ([/Users/rolandklees/miniforge3/envs/stsjax/lib/python3.11/site-packages/dynamax/types.py](http://localhost:8890/Users/rolandklees/miniforge3/envs/stsjax/lib/python3.11/site-packages/dynamax/types.py)) I also suffer from...
