diff --git a/checkpoint/orbax/checkpoint/experimental/v1/path.py b/checkpoint/orbax/checkpoint/experimental/v1/path.py
index 148979552..6c9b55ede 100644
--- a/checkpoint/orbax/checkpoint/experimental/v1/path.py
+++ b/checkpoint/orbax/checkpoint/experimental/v1/path.py
@@ -21,3 +21,4 @@
     PathLike,
     PathAwaitingCreation,
 )
+from orbax.checkpoint.experimental.v1._src.path import step
diff --git a/docs/api_reference/checkpoint.v1.path.rst b/docs/api_reference/checkpoint.v1.path.rst
index 6cccb4f7f..e09859399 100644
--- a/docs/api_reference/checkpoint.v1.path.rst
+++ b/docs/api_reference/checkpoint.v1.path.rst
@@ -14,3 +14,8 @@ Types
   :members:
 .. autoclass:: PathAwaitingCreation
   :members:
+
+Step entities
+------------------------------------------------------------
+.. automodule:: orbax.checkpoint.experimental.v1.path.step
+  :members: