numpy>=1.19.5
gdown>=4.4.0
tensorflow==2.6.2
tensorflow_io==0.21.0
tensorflow_probability==0.14.1
tensorflow_addons==0.15.0
