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