tensorflow==2.9.0
tensorflow_probability==0.17.0
numpy