diff --git a/CHANGELOG.md b/CHANGELOG.md
index 2d2aac354..aa3c8aae8 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -8,6 +8,7 @@ changes. Best viewed [here](https://google-grain.readthedocs.io/en/latest/change
 * New features:
   * Adds automated batching into shared memory when multiprocess prefetch is
     used to save one data copy.
+  * Adds experimental `ShapeDtypeStruct` to represent dataset element specs.
 
 * Breaking changes:
 
diff --git a/grain/experimental.py b/grain/experimental.py
index 297c40b56..efcd81d68 100644
--- a/grain/experimental.py
+++ b/grain/experimental.py
@@ -27,6 +27,7 @@
 from grain._src.python.dataset.base import (
     DatasetOptions,
     ExecutionTrackingMode,
+    ShapeDtypeStruct,
 )
 from grain._src.python.dataset.dataset import (
     apply_transformations,