Once https://github.com/probml/sts-jax is released we could add the Causal Impact implementation in [sts-jax](https://github.com/probml/sts-jax).