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