Update to new PT Theme

stack-info: PR: #2361, branch: drisspg/stack/76

Author: drisspg

docs/README.md

@@ -0,0 +1,65 @@
+# torchao Documentation
+
+This directory contains the source files for the torchao documentation.
+
+## Building the Documentation
+
+### Prerequisites
+
+Install the required dependencies:
+
+```bash
+pip install -r requirements.txt
+pip install sphinx-serve
+```
+
+### Build the Documentation
+
+To build the HTML documentation:
+
+```bash
+make html
+```
+
+The built documentation will be in the `build/html` directory.
+
+### Serve the Documentation Locally
+
+After building, you can serve the documentation locally using `sphinx-serve`:
+
+```bash
+sphinx-serve -b build
+```
+
+This will start a local server (typically at http://localhost:8000) where you can view the documentation with live reload capabilities.
+
+## Documentation Structure
+
+- `source/` - Source files for the documentation
+  - `conf.py` - Sphinx configuration file
+  - `index.rst` - Main documentation entry point
+  - `tutorials.rst` - Tutorials section index
+  - `_static/` - Static files (CSS, images, etc.)
+  - `_templates/` - Custom templates
+  - `tutorials_source/` - Executable tutorial Python files (for sphinx-gallery)
+  - `tutorials/` - Generated tutorial gallery (auto-generated, don't edit)
+- `build/` - Generated documentation output (created after building)
+
+## Tutorial Types
+
+This documentation has two types of tutorials:
+
+### 1. Static Tutorials
+Educational content, guides, and explanations that are written as `.rst` or `.md` files:
+- Located directly in `source/` (e.g., `serialization.rst`, `subclass_basic.rst`)
+- Referenced in `source/tutorials.rst`
+- These are traditional documentation pages
+
+### 2. Executable Tutorials
+Interactive code examples and demos that can be run:
+- Source files: `source/tutorials_source/*.py` (Python scripts with special formatting)
+- Generated output: `source/tutorials/` (auto-generated HTML gallery)
+- Built using sphinx-gallery extension
+- Each Python file becomes a downloadable notebook and HTML page
+
+When you run `make html`, sphinx-gallery automatically converts Python files in `tutorials_source/` into an interactive gallery in the `tutorials/` directory.

docs/README.txt

@@ -1,9 +1,14 @@
-sphinx-gallery>0.11
-sphinx==5.0.0
-sphinx_design
-sphinx_copybutton
-sphinx-tabs
-matplotlib
+sphinx==5.3.0
+-e git+https://github.com/pytorch/pytorch_sphinx_theme.git@pytorch_sphinx_theme2#egg=pytorch_sphinx_theme2
+sphinxcontrib.katex==0.8.6
+sphinxext-opengraph==0.9.1
+docutils==0.17.1  # Changed from 0.16 to match sphinx-tabs requirement
+sphinx-design==0.4.0
+sphinxcontrib-mermaid==1.0.0
+myst-parser==0.18.1  # if want to contribute in markdown
+sphinx-gallery==0.11.1  # if hosting interactive tutorials
+sphinx-tabs==3.4.0
+sphinx-copybutton==0.5.2
+sphinx-sitemap==2.6.0
 myst-parser
 sphinxcontrib-mermaid==1.0.0
--e git+https://github.com/pytorch/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme

docs/requirements.txt

@@ -185,3 +185,40 @@ article.pytorch-article .sphx-glr-thumbnails .sphx-glr-thumbcontainer {
 article.pytorch-article div.section div.wy-table-responsive tbody td {
     width: 50%;
 }
+
+/* Mermaid diagram styling */
+.mermaid {
+    text-align: center;
+    margin: 1.5em 0;
+}
+
+/* Ensure mermaid diagrams are visible */
+div.mermaid {
+    display: block !important;
+    width: 100% !important;
+    min-height: 200px;
+    background: #f8f9fa;
+    border: 1px solid #dee2e6;
+    border-radius: 0.25rem;
+    padding: 1rem;
+}
+
+/* Fix for mermaid in pytorch theme */
+.pytorch-article .mermaid {
+    overflow: visible !important;
+}
+
+/* Fix for navbar overlapping content */
+.bd-container {
+    padding-top: 1rem !important;
+}
+
+/* Add proper spacing between navbar and content */
+.bd-article-container {
+    margin-top: 1rem;
+}
+
+/* Ensure proper spacing for the main content area */
+article.bd-article {
+    padding-top: 1rem;
+}

docs/source/_static/css/custom.css

@@ -0,0 +1,12 @@
+API Reference
+=============
+
+.. toctree::
+   :maxdepth: 2
+
+   api_ref_intro
+   api_ref_dtypes
+   api_ref_quantization
+   api_ref_sparsity
+   api_ref_float8
+   api_ref_kernel