Fitting a local level model using the Tensorflow Probability

Question

I have simulated time-series data with pre-defined parameters below. I want to build a local linear model that approximates the true mean at 5.

# Set true par