tensorflow>=2.1.0
tensorflow-addons>=0.7.0
tensorflow_probability>=0.9
numpy
six >= 1.10.0