tensorflow==2.1.0 tensorflow_probability=0.7.0 matplotlib==3.1.2 numpy==1.17.5