[BE/docs] Add float8 training api ref to docsite (pytorch#2313)

danielvegamyhre · web-flow · commit 72ea2fc5f085 · 2025-06-05T10:43:11.000-04:00
add float8 training api to api ref docs
diff --git a/docs/source/api_ref_float8.rst b/docs/source/api_ref_float8.rst
@@ -0,0 +1,30 @@
+.. _api_quantization:
+
+====================
+torchao.float8
+====================
+
+.. currentmodule:: torchao.float8
+
+Main float8 training APIs
+----------------------
+
+.. autosummary::
+    :toctree: generated/
+    :nosignatures:
+
+    convert_to_float8_training
+
+Other float8 training types
+-------------------------------
+
+.. autosummary::
+    :toctree: generated/
+    :nosignatures:
+
+    Float8LinearConfig
+    Float8GEMMConfig
+    CastConfig
+    ScalingType
+    ScalingGranularity
+    precompute_float8_dynamic_scale_for_fsdp
diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -30,6 +30,7 @@ for an overall introduction to the library and recent highlight and updates.
    api_ref_dtypes
    api_ref_quantization
    api_ref_sparsity
+   api_ref_float8
 
 .. toctree::
    :glob:
