Keras 3 breaks Tensorflow Probability upon import about probability HOT 4 CLOSED

We plan to release TensorFlow Probability 0.24.0 a week or so after TensorFlow 2.16.0 is released.

I don't know when the TensorFlow team plans to release TF 2.16.0. (It looks like TF 2.16.0-rc0 was made available last week.)

Until then, you should be able to import tensorflow_probabability with Keras 3 installed if you also install tfp-nightly and tf-nightly (and tf-keras-nightly).

from probability.

TensorFlow Probability 0.23.0 is not compatible with Keras 3.

And it looks like TensorFlow 2.15.0 is also not compatible with Keras 3. When I run those pip commands, upgrading Keras gives the error message:

tensorflow 2.15.0 requires keras<2.16,>=2.15.0, but you have keras 3.0.0 which is incompatible.

It will be possible to import TensorFlow Probability 0.24.0 with TensorFlow 2.16.0 and Keras 3 installed. But please note that TensorFlow Probability 0.24.0 will continue to use Keras 2, which means that tf-keras 2.16.0 will also have to be installed.

from probability.

@jburnim Do you have a rough timeline for when we can expect the fix to be released? If anyone knows of a work-around I would also be grateful.

from probability.

https://stackoverflow.com/a/78233592/13086128

from probability.