tensorflow==2.9.0
tensorflow_probability
numpy