diff --git a/arraycontext/__init__.py b/arraycontext/__init__.py
index a338059d1caa3e27d680321b5e4c58da51308aa3..8562f7b7a70eac986187ffbe517d17a9899c9284 100644
--- a/arraycontext/__init__.py
+++ b/arraycontext/__init__.py
@@ -52,6 +52,7 @@ from .container.traversal import (
         rec_multimap_array_container,
         mapped_over_array_containers,
         multimapped_over_array_containers,
+        keyed_map_array_container, rec_keyed_map_array_container,
         thaw, freeze,
         from_numpy, to_numpy)
 
@@ -83,6 +84,7 @@ __all__ = (
         "rec_map_array_container", "rec_multimap_array_container",
         "mapped_over_array_containers",
         "multimapped_over_array_containers",
+        "keyed_map_array_container", "rec_keyed_map_array_container",
         "thaw", "freeze",
         "from_numpy", "to_numpy",
 
diff --git a/arraycontext/container/traversal.py b/arraycontext/container/traversal.py
index 1dc0d3b1193bea0df29baba96dada561f10f10e9..0bd0a1daae7c6c093e661c93152956e3b2df539b 100644
--- a/arraycontext/container/traversal.py
+++ b/arraycontext/container/traversal.py
@@ -7,6 +7,8 @@
 .. autofunction:: multimap_array_container
 .. autofunction:: rec_map_array_container
 .. autofunction:: rec_multimap_array_container
+.. autofunction:: keyed_map_array_container
+.. autofunction:: rec_keyed_map_array_container
 
 Traversing decorators
 ~~~~~~~~~~~~~~~~~~~~~