diff --git a/docs/api_reference.rst b/docs/api_reference.rst
index 412ffc9..2586e02 100644
--- a/docs/api_reference.rst
+++ b/docs/api_reference.rst
@@ -44,6 +44,7 @@ argument to :class:`~dpsynth.TabularConfig`.
    :nosignatures:
    :template: autosummary/class.rst
 
+   Schema
    CategoricalAttribute
    NumericalAttribute
    OpenSetCategoricalAttribute
diff --git a/docs/conf.py b/docs/conf.py
index c15e515..a7a0782 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -45,9 +45,22 @@
     'show-inheritance': True,
 }
 
-# templates_path = ['_templates']
+templates_path = ['_templates']
 exclude_patterns = ['_build', 'Thumbs.db', '.DS_Store']
 
+root_doc = 'index'
+master_doc = 'index'
+
+autodoc_mock_imports = [
+    'pipeline_dp',
+    'apache_beam',
+    'tensorflow',
+    'flax',
+    'gemma',
+    'jax_privacy',
+    'optax',
+]
+
 # -- Options for HTML output -------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#options-for-html-output
 
diff --git a/docs/in_memory_api.md b/docs/in_memory_api.md
index 5a16d3e..8e2c3b8 100644
--- a/docs/in_memory_api.md
+++ b/docs/in_memory_api.md
@@ -53,6 +53,9 @@ When initializing `dpsynth.TabularConfig`:
     partition selection (default: `0.1`).
 *   `cross_attribute_constraints`: Optional sequence of constraints to enforce
     on generated data.
+*   `compress_columns`: Whether to compress rare categories (< 3*sigma) for
+    `CategoricalAttribute` columns not present in constraints
+    (default: `False`).
 
 When calling `config.calibrate(...)`:
 
diff --git a/docs/index.md b/docs/index.md
index 3576de8..7eb2bec 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -136,7 +136,7 @@ APIs:
 
 data_and_terminology
 in_memory_api
-scalable_pipeline_api
+scalable_beam_api
 ```
 
 ```{toctree}
diff --git a/docs/processing_lifecycle.md b/docs/processing_lifecycle.md
index d712757..99364ed 100644
--- a/docs/processing_lifecycle.md
+++ b/docs/processing_lifecycle.md
@@ -11,7 +11,8 @@ floating-point ranges, unbounded string categories, high-dimensional state
 spaces, and strict privacy budget accounting.
 
 This page details the 5-stage mathematical lifecycle executed across both the
-[In-Memory](in_memory_api.md) and [Scalable Pipeline](scalable_beam_api.md)
+[In-Memory](in_memory_api.md) and
+[Scalable Pipeline](scalable_beam_api.md)
 APIs.
 
 ```