diff --git a/README.md b/README.md
index f555346..929345f 100644
--- a/README.md
+++ b/README.md
@@ -9,11 +9,12 @@ goal.
 
 Currently we use:
 
-*   flax for modules
-*   optax for optimization
-*   orbax for checkpointing
-*   pygrain for data
-*   ConfigDict for hyper-parameters.
+*   [flax](https://github.com/google/flax) for modules
+*   [optax](https://github.com/google-deepmind/optax) for optimization
+*   [orbax](https://github.com/google/orbax) for checkpointing
+*   [tfds](https://github.com/tensorflow/datasets) for data
+*   [pygrain](https://github.com/google/grain) for data loading
+*   [ConfigDict](https://github.com/google/ml_collections) for hyper-parameters.
 
 Not currently supported or left out for simplicity:
 
diff --git a/pyproject.toml b/pyproject.toml
index 020d60d..c29a72a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -60,8 +60,7 @@ dependencies = [
     "optax>=0.2.2",
     "orbax>=0.1.7",
     "sentencepiece>=0.2.0",
-    # TODO: Temporary fix while waiting for TFDS to be released.
-    "tfds-nightly",
+    "tensorflow_datasets>=4.9.5",
     "tensorflow>=2.16.1",
 ]