docs: python api reference (#109)

<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

* **New Features**
  * Integrated Vercel Analytics for usage insights.
* Automated generation and publication of Python API docs to the web
app.
  * Improved docs routing with targeted caching for faster load times.

* **Documentation**
* Added a comprehensive Python API reference covering apply, tracing,
and schema modules.

* **Chores**
* Updated build/prebuild flow and packaging to produce and serve docs;
added doc build requirements and new ignore patterns.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

Author: xslingcn

.gitignore

@@ -38,6 +38,10 @@ MANIFEST
 !web/**/lib/
 !web/**/lib/**
 
+# Sphinx
+.sphinx-deps
+web/**/public
+
 .claude
 .vscode
 

docs/api/_static/brand/fib-dark.png

@@ -0,0 +1,136 @@
+import os
+import sys
+from datetime import datetime
+from pathlib import Path
+from typing import Any, List
+
+import tomli
+
+# import tlcpack_sphinx_addon
+# Configuration file for the Sphinx documentation builder.
+#
+# For the full list of built-in configuration values, see the documentation:
+# https://www.sphinx-doc.org/en/master/usage/configuration.html
+
+# -- Project information -----------------------------------------------------
+# https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
+
+repo_root = Path(__file__).resolve().parents[2]
+sys.path.insert(0, str(repo_root))
+os.environ["BUILD_DOC"] = "1"
+
+project = "FlashInfer-Bench"
+author = "FlashInfer Community"
+copyright = f"{datetime.now().year}, {author}"
+
+try:
+    from setuptools_scm import get_version as scm_get_version
+except Exception:
+    scm_get_version = None
+
+
+def _compute_version() -> str:
+    v = os.environ.get("FLASHINFER_BENCH_VERSION")
+    if v:
+        return v
+    if scm_get_version is not None:
+        try:
+            return scm_get_version(root=str(repo_root), fallback_version="0.0.0")
+        except Exception:
+            pass
+    try:
+        with open(repo_root / "pyproject.toml", "rb") as f:
+            data = tomli.load(f)
+        v = data.get("project", {}).get("version")
+        if v:
+            return v
+    except Exception:
+        pass
+    return "0.0.0+docs"
+
+
+__version__ = _compute_version()
+
+
+# -- General configuration ---------------------------------------------------
+# https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration
+
+extensions = [
+    "myst_parser",
+    "sphinx_tabs.tabs",
+    "sphinx.ext.autodoc",
+    "sphinx.ext.napoleon",
+    "sphinx.ext.autosummary",
+    "sphinx.ext.mathjax",
+    "sphinxcontrib.autodoc_pydantic",
+]
+
+autodoc_mock_imports = [
+    "torch",
+    "triton",
+    "flashinfer._build_meta",
+    "cuda",
+    "numpy",
+    "einops",
+    "mpi4py",
+    "safetensors",
+]
+autodoc_default_flags = ["members"]
+autodoc_class_signature = "separated"
+autodoc_member_order = "bysource"
+autodoc_default_options = {"exclude-members": "model_config"}
+autodoc_typehints = "both"
+
+autodoc_pydantic_model_show_validator_summary = False
+autodoc_pydantic_model_show_validator_members = False
+autodoc_pydantic_model_show_config_summary = False
+autodoc_pydantic_field_list_validators = False
+autodoc_pydantic_model_summary_list_order = "bysource"
+autodoc_pydantic_model_member_order = "bysource"
+
+autosummary_generate = True
+
+source_suffix = [".rst", ".md"]
+
+language = "en"
+
+exclude_patterns = ["_build", "Thumbs.db", ".DS_Store", ".sphinx-deps", ".sphinx-deps/**"]
+
+# The name of the Pygments (syntax highlighting) style to use.
+pygments_style = "sphinx"
+
+# A list of ignored prefixes for module index sorting.
+# If true, `todo` and `todoList` produce output, else they produce nothing.
+todo_include_todos = False
+
+myst_enable_extensions = [
+    "dollarmath",
+    "amsmath",
+    "deflist",
+    "colon_fence",
+    "html_image",
+    "linkify",
+    "substitution",
+]
+
+myst_heading_anchors = 3
+myst_ref_domains = ["std", "py"]
+myst_all_links_external = False
+
+# -- Options for HTML output ----------------------------------------------
+
+html_theme = "shibuya"
+
+templates_path: List[Any] = ["_templates"]
+
+html_static_path = ["_static"]
+
+
+html_theme_options = {
+    "accent_color": "indigo",
+    "logo_target": "/",
+    "light_logo": "_static/brand/fib-light.png",
+    "dark_logo": "_static/brand/fib-dark.png",
+    "nav_links": [{"title": "Docs", "url": f"/docs", "external": True}],
+    "globaltoc_expand_depth": 1,
+}

docs/api/_static/brand/fib-light.png

@@ -0,0 +1,10 @@
+# FlashInfer-Bench Python API
+
+```{toctree}
+:maxdepth: 2
+:caption: API Reference
+
+rst/apply
+rst/tracing
+rst/schema
+```

docs/api/conf.py

@@ -0,0 +1,8 @@
+autodoc_pydantic>=2.1,<3
+myst-parser[linkify]>=4,<5
+setuptools-scm>=8
+shibuya
+sphinx==7.4.*
+sphinx-tabs>=3.4.7
+sphinx-toolbox
+tomli

docs/api/index.md

@@ -0,0 +1,15 @@
+flashinfer_bench.apply
+======================
+
+.. currentmodule:: flashinfer_bench
+
+``flashinfer_bench.apply`` provides a tool that meets two needs:
+
+1. **Apply** best-performing one from FlashInfer Trace database to the LLM engine
+2. **Trace** the kernel in the LLM engine and dump its input as FlashInfer Trace's workload format
+
+.. autofunction:: apply
+
+.. autofunction:: enable_apply
+
+.. autofunction:: disable_apply

docs/api/requirements.txt

@@ -0,0 +1,18 @@
+FlashInfer Trace Schema Python API
+==================================
+
+FlashInfer-Bench provides a schema for the FlashInfer Trace database. This document includes
+the Python API for the schema, including
+
+- The :class:`Definition` class, which defines the kernel specification.
+- The :class:`Solution` class, which defines the kernel solution.
+- The :class:`Trace` class, which defines the kernel execution trace.
+- The :class:`TraceSet` class, which defines a set of kernel execution traces.
+
+.. toctree::
+    :maxdepth: 2
+
+    schema_definition
+    schema_solution
+    schema_trace
+    schema_traceset

docs/api/rst/apply.rst

@@ -0,0 +1,15 @@
+Definition
+==========
+
+.. currentmodule:: flashinfer_bench.data
+
+.. autopydantic_model:: Definition
+
+.. autopydantic_model:: AxisConst
+
+.. autopydantic_model:: AxisVar
+
+.. autopydantic_model:: TensorSpec
+
+.. autoclass:: flashinfer_bench.data.definition.DType
+    :members:

docs/api/rst/schema.rst

@@ -0,0 +1,13 @@
+Solution
+========
+
+.. currentmodule:: flashinfer_bench.data
+
+.. autopydantic_model:: Solution
+
+.. autoclass:: SupportedLanguages
+    :members:
+
+.. autopydantic_model:: SourceFile
+
+.. autopydantic_model:: BuildSpec