tensorflow / probability

Probabilistic reasoning and statistical analysis in TensorFlow

Home Page:https://www.tensorflow.org/probability/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

`_parameter_properties` is not implemented for `LinearGaussianStateSpaceModel`

Joshuaalbert opened this issue · comments

When using the JAX substrate doing optimisation of the parameters of LinearGaussianStateSpaceModel when I JIT compile the update step of optimisation I get the below at this line: https://github.com/tensorflow/probability/blob/v0.23.0/tensorflow_probability/python/internal/auto_composite_tensor.py#L795

`_parameter_properties` is not implemented for `LinearGaussianStateSpaceModel`