diff --git a/requirements.txt b/requirements.txt
index 470bd0ef..53141c10 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,3 +8,5 @@ tensorflow-text==1.15.0rc0
 tensor2tensor==1.15.0
 tfds-nightly
 tensorflow-gpu==1.15.2
+tensorflow-datasets==3.0.0
+tfds-nightly==3.1.0.dev202006230105