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