Add stable API references (#2399)

Summary: Pull Request resolved: #2399 Differential Revision: D62700154
pytorch · Sep 15, 2024 · 9d5c872 · 9d5c872
1 parent e75f376
commit 9d5c872
Show file tree

Hide file tree

Showing 8 changed files with 23 additions and 1 deletion.
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -1,4 +1,5 @@
 sphinx==5.0.0
+myst_parser
 pyre-extensions
 # torch
 # PyTorch Theme

diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -49,7 +49,11 @@
 # Add any Sphinx extension module names here, as strings. They can be
 # extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
 # ones.
-extensions = ["sphinx.ext.napoleon", "sphinx.ext.autodoc"]
+extensions = [
+    "sphinx.ext.napoleon",
+    "sphinx.ext.autodoc",
+    "myst_parser",
+]
 
 # Add any paths that contain templates here, relative to this directory.
 templates_path = ["_templates"]

diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -35,6 +35,17 @@ TorchRec API
 
    overview.rst
 
+.. toctree::
+   :maxdepth: 1
+   :caption: Stable API References
+   :hidden:
+
+   stable/datatypes-api-reference.rst
+   stable/modules-api-reference.rst
+   stable/planner-api-reference.rst
+   stable/model-parallel-api-reference.rst
+   stable/inference-api-reference.rst
+
 .. toctree::
    :maxdepth: 1
    :caption: All API References

diff --git a/docs/source/stable/datatypes-api-reference.rst b/docs/source/stable/datatypes-api-reference.rst
@@ -0,0 +1,6 @@
+TorchRec DataTypes API Reference
+----------------------------------
+
+.. autoclass:: JaggedTensor
+.. autoclass:: KeyedJaggedTensor
+.. autoclass:: KeyedTensor
diff --git a/docs/source/stable/inference-api-reference.rst b/docs/source/stable/inference-api-reference.rst
diff --git a/docs/source/stable/model-parallel-api-reference.rst b/docs/source/stable/model-parallel-api-reference.rst
diff --git a/docs/source/stable/modules-api-reference.rst b/docs/source/stable/modules-api-reference.rst
diff --git a/docs/source/stable/planner-api-reference.rst b/docs/source/stable/planner-api-reference.rst