Update requirements

PiperOrigin-RevId: 413947906

requirements.txt

@@ -16,6 +16,7 @@ flatbuffers==1.12
 flax==0.3.6
 future==0.18.2
 gast==0.4.0
+gcsfs==2021.11.1
 gin==0.1.6
 gin-config==0.5.0
 google-auth==1.35.0
@@ -64,6 +65,7 @@ tensorboard==2.6.0
 tensorboard-data-server==0.6.1
 tensorboard-plugin-wit==1.8.0
 tensorflow==2.7.0rc1
+tensorflow-datasets==4.4.0
 tensorflow-estimator==2.7.0
 tensorflow-io-gcs-filesystem==0.21.0
 tensorflow-probability==0.14.1
@@ -78,3 +80,4 @@ urllib3==1.26.7
 Werkzeug==2.0.2
 wrapt==1.12.1
 zipp==3.6.0
+zarr==2.10.3