Add API documentation using autodoc and autosummary

.gitignore

@@ -18,6 +18,9 @@ scratch/
 *.egg
 examples/tensorflow/distributed_mnist/data
 *.iml
+
+# Sphinx documentation
+docs/_build/
 doc/_build
 doc/_static
 doc/_templates

.readthedocs.yaml

@@ -0,0 +1,20 @@
+version: 2
+
+build:
+  os: ubuntu-22.04
+  tools:
+    python: "3.10"
+
+python:
+  install:
+    - method: pip
+      path: .
+    - requirements: docs/requirements.txt
+
+sphinx:
+  configuration: docs/conf.py
+  fail_on_warning: false
+
+formats:
+  - pdf
+  - epub

docs/Makefile

@@ -0,0 +1,12 @@
+SPHINXOPTS    ?=
+SPHINXBUILD   ?= sphinx-build
+SOURCEDIR     = .
+BUILDDIR      = _build
+
+help:
+	@$(SPHINXBUILD) -M help "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)
+
+.PHONY: help Makefile
+
+%: Makefile
+	@$(SPHINXBUILD) -M $@ "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)

docs/_static/custom.css

@@ -0,0 +1,345 @@
+/* What's New Section Styling */
+.whats-new-container {
+    display: grid;
+    grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
+    gap: 20px;
+    margin: 30px 0;
+}
+
+.new-feature-card {
+    background: linear-gradient(135deg, #f8f9fa 0%, #e9ecef 100%);
+    border: 1px solid #dee2e6;
+    border-radius: 12px;
+    padding: 25px;
+    box-shadow: 0 4px 12px rgba(0, 0, 0, 0.1);
+    transition: transform 0.2s ease, box-shadow 0.2s ease;
+    position: relative;
+    overflow: hidden;
+}
+
+.new-feature-card:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 8px 25px rgba(0, 0, 0, 0.15);
+}
+
+.new-feature-card.exclusive {
+    background: linear-gradient(135deg, #ff6b6b 0%, #4ecdc4 100%);
+    color: white;
+    border: 2px solid #ff4757;
+}
+
+.new-feature-card.exclusive::before {
+    content: "V3 EXCLUSIVE";
+    position: absolute;
+    top: 10px;
+    right: -30px;
+    background: #ff4757;
+    color: white;
+    padding: 5px 40px;
+    font-size: 0.75em;
+    font-weight: bold;
+    transform: rotate(45deg);
+    box-shadow: 0 2px 5px rgba(0, 0, 0, 0.2);
+}
+
+.feature-icon {
+    font-size: 2.5em;
+    margin-bottom: 15px;
+    display: block;
+}
+
+.new-feature-card h3 {
+    margin-top: 0;
+    margin-bottom: 15px;
+    font-size: 1.3em;
+    font-weight: 600;
+}
+
+.new-feature-card.exclusive h3 {
+    color: white;
+}
+
+.new-feature-card ul {
+    margin: 15px 0;
+    padding-left: 20px;
+}
+
+.new-feature-card li {
+    margin-bottom: 8px;
+    line-height: 1.5;
+}
+
+.new-feature-card code {
+    background: rgba(0, 0, 0, 0.1);
+    padding: 2px 6px;
+    border-radius: 4px;
+    font-size: 0.9em;
+}
+
+.new-feature-card.exclusive code {
+    background: rgba(255, 255, 255, 0.2);
+    color: white;
+}
+
+/* Collapse navigation tree items by default */
+.bd-toc .toctree-l1 > ul {
+    display: none;
+}
+
+.bd-toc .toctree-l1.current > ul {
+    display: block;
+}
+
+.bd-toc .toctree-l2 > ul {
+    display: none;
+}
+
+.bd-toc .toctree-l2.current > ul {
+    display: block;
+}
+
+/* Highlight model customization in navigation */
+.bd-toc a[href*="model_customization"], 
+.toctree-l1 a[href*="model_customization"] {
+    background: linear-gradient(90deg, rgba(255, 107, 107, 0.15), rgba(78, 205, 196, 0.15));
+    border-left: 4px solid #ff6b6b;
+    padding-left: 12px;
+    font-weight: 600;
+    border-radius: 4px;
+    margin: 2px 0;
+    display: block;
+}
+
+.bd-toc a[href*="model_customization"]:before,
+.toctree-l1 a[href*="model_customization"]:before {
+    content: "⭐ ";
+    color: #ff4757;
+    font-size: 0.9em;
+}
+
+/* Feedback Component */
+.feedback-container {
+    border-top: 1px solid #e1e5e9;
+    margin-top: 40px;
+    padding: 24px;
+    background-color: #f8f9fa;
+    border-radius: 8px;
+    border: 1px solid #e1e5e9;
+}
+
+.feedback-question {
+    font-size: 20px;
+    font-weight: 500;
+    color: #374151;
+    margin-bottom: 20px;
+    text-align: left;
+}
+
+.feedback-buttons {
+    display: flex;
+    justify-content: flex-start;
+    gap: 12px;
+}
+
+.feedback-btn {
+    display: flex;
+    align-items: center;
+    gap: 8px;
+    padding: 8px 16px;
+    border: 1px solid #d1d5db;
+    border-radius: 6px;
+    background: white;
+    color: #6b7280;
+    font-size: 14px;
+    cursor: pointer;
+    transition: all 0.2s ease;
+}
+
+.feedback-btn:hover {
+    border-color: #9ca3af;
+    background: #f9fafb;
+}
+
+.feedback-btn.selected {
+    border-color: #3b82f6;
+    background: #eff6ff;
+    color: #1d4ed8;
+}
+
+.feedback-btn svg {
+    width: 16px;
+    height: 16px;
+}
+
+.feedback-thanks {
+    display: none;
+    color: #059669;
+    font-size: 14px;
+    text-align: center;
+    margin-top: 16px;
+}
+
+/* Feedback Options */
+.feedback-options {
+    display: none;
+    margin-top: 20px;
+}
+
+.feedback-options-title {
+    font-size: 16px;
+    font-weight: 500;
+    color: #374151;
+    margin-bottom: 16px;
+}
+
+.required {
+    color: #ff6b6b;
+}
+
+.feedback-checkboxes {
+    display: flex;
+    flex-direction: column;
+    gap: 12px;
+    margin-bottom: 20px;
+}
+
+.feedback-checkbox {
+    display: flex;
+    align-items: center;
+    cursor: pointer;
+    color: #374151;
+    font-size: 14px;
+    position: relative;
+    padding-left: 32px;
+}
+
+.feedback-checkbox input[type="checkbox"] {
+    position: absolute;
+    opacity: 0;
+    cursor: pointer;
+    height: 0;
+    width: 0;
+}
+
+.checkmark {
+    position: absolute;
+    left: 0;
+    top: 2px;
+    height: 18px;
+    width: 18px;
+    background-color: transparent;
+    border: 2px solid #6b7280;
+    border-radius: 3px;
+}
+
+.feedback-checkbox:hover input ~ .checkmark {
+    border-color: #9ca3af;
+}
+
+.feedback-checkbox input:checked ~ .checkmark {
+    background-color: #3b82f6;
+    border-color: #3b82f6;
+}
+
+.checkmark:after {
+    content: "";
+    position: absolute;
+    display: none;
+}
+
+.feedback-checkbox input:checked ~ .checkmark:after {
+    display: block;
+}
+
+.feedback-checkbox .checkmark:after {
+    left: 5px;
+    top: 2px;
+    width: 4px;
+    height: 8px;
+    border: solid white;
+    border-width: 0 2px 2px 0;
+    transform: rotate(45deg);
+}
+
+.feedback-more {
+    display: flex;
+    flex-direction: column;
+    gap: 12px;
+}
+
+.feedback-more-title {
+    font-size: 16px;
+    font-weight: 500;
+    color: #374151;
+}
+
+.feedback-textarea {
+    padding: 12px;
+    border: 1px solid #d1d5db;
+    border-radius: 6px;
+    font-size: 14px;
+    resize: vertical;
+    min-height: 80px;
+    font-family: inherit;
+    background-color: #ffffff;
+    color: #374151;
+}
+
+.feedback-textarea::placeholder {
+    color: #6b7280;
+}
+
+.feedback-textarea:focus {
+    outline: none;
+    border-color: #3b82f6;
+    box-shadow: 0 0 0 3px rgba(59, 130, 246, 0.1);
+}
+
+.feedback-submit-btn {
+    padding: 10px 20px;
+    background: #3b82f6;
+    color: white;
+    border: none;
+    border-radius: 6px;
+    font-size: 14px;
+    font-weight: 500;
+    cursor: pointer;
+    align-self: flex-start;
+    transition: background 0.2s;
+}
+
+.feedback-submit-btn:hover {
+    background: #2563eb;
+}
+
+header {
+    background-color: white;
+
+    box-shadow: 0 1px 2px rgba(0, 0, 0, 0.05);
+    position: sticky;
+    top: 0;
+    z-index: 50;
+}
+
+h1 {
+    font-size: 1.875rem;
+    font-weight: 700;
+    color: #111827;
+}
+
+h2 {
+    font-size: 1.5rem;
+    font-weight: 700;
+    color: #111827;
+}
+
+h3 {
+    font-size: 1.25rem;
+    font-weight: 500;
+    color: #111827;
+}
+
+p {
+    font-size: 1.0rem;
+    color: #4b5563;
+}