tensorflow==2.0.0
tensorflow-addons==0.6.0
tensorflow_probability==0.8
numpy
six>=1.10.0
