tensorflow==2.2.0
tensorflow-datasets==3.1.0
tensorflow-probability==0.10.0