dm-sonnet==1.36
tensorflow-probability==0.8.0
protobuf<3.20
gast==0.2.2
h5py==2.10.0
numpy>=1.16,<1.19
