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