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

[tf]
tensorflow>=1.0.0

[tf_gpu]
tensorflow-gpu>=1.0.0
