search

tensorflow / probability

Probabilistic reasoning and statistical analysis in TensorFlow
https://www.tensorflow.org/probability/
Apache License 2.0
4.25k stars 1.1k forks source link

`_parameter_properties` is not implemented for `LinearGaussianStateSpaceModel` #1794

Open Joshuaalbert opened 6 months ago

Joshuaalbert commented 6 months ago

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`