absl-py~=1.0.0
attrs~=21.2.0
dm-tree~=0.1.1
matplotlib~=3.3.4
mpmath
scipy>=0.17
tensorflow-datasets>=4.4.0
tensorflow-estimator>=2.3.0
tensorflow-probability>=0.13.0
tensorflow>=1.14