diff --git a/docs/api-assorted.md b/docs/api-assorted.md
index 26ccb4d7..6d78e26f 100644
--- a/docs/api-assorted.md
+++ b/docs/api-assorted.md
@@ -1,40 +1,132 @@
 # Assorted functions
 
+## Creation functions
+
 ```{eval-rst}
 .. currentmodule:: array_api_extra
 .. autosummary::
     :nosignatures:
     :toctree: generated
 
-    angle
-    apply_where
-    argpartition
-    at
-    atleast_nd
-    broadcast_shapes
-    cov
     create_diagonal
+    one_hot
+```
+
+## Inspection
+
+```{eval-rst}
+.. currentmodule:: array_api_extra
+.. autosummary::
+    :nosignatures:
+    :toctree: generated
+
     default_dtype
+```
+
+## Element-wise functions
+
+```{eval-rst}
+.. currentmodule:: array_api_extra
+.. autosummary::
+    :nosignatures:
+    :toctree: generated
+
+    angle
+    apply_where
     deg2rad
-    diag_indices
-    expand_dims
     isclose
-    isin
-    kron
     nan_to_num
-    nanmax
-    nanmin
-    nansum
-    nunique
-    one_hot
-    pad
-    partition
     rad2deg
-    searchsorted
-    setdiff1d
     sinc
+```
+
+## Indexing functions
+
+```{eval-rst}
+.. currentmodule:: array_api_extra
+.. autosummary::
+    :nosignatures:
+    :toctree: generated
+
+    at
+    diag_indices
     tril_indices
     triu_indices
-    union1d
     unravel_index
 ```
+
+## Linear algebra functions
+
+```{eval-rst}
+.. currentmodule:: array_api_extra
+.. autosummary::
+    :nosignatures:
+    :toctree: generated
+
+    kron
+```
+
+## Manipulation functions
+
+```{eval-rst}
+.. currentmodule:: array_api_extra
+.. autosummary::
+    :nosignatures:
+    :toctree: generated
+
+    atleast_nd
+    broadcast_shapes
+    expand_dims
+    pad
+```
+
+## Searching functions
+
+```{eval-rst}
+.. currentmodule:: array_api_extra
+.. autosummary::
+    :nosignatures:
+    :toctree: generated
+
+    searchsorted
+```
+
+## Set functions
+
+```{eval-rst}
+.. currentmodule:: array_api_extra
+.. autosummary::
+    :nosignatures:
+    :toctree: generated
+
+    isin
+    nunique
+    setdiff1d
+    union1d
+```
+
+## Sorting functions
+
+```{eval-rst}
+.. currentmodule:: array_api_extra
+.. autosummary::
+    :nosignatures:
+    :toctree: generated
+
+    argpartition
+    partition
+```
+
+## Statistical functions
+
+```{eval-rst}
+.. currentmodule:: array_api_extra
+.. autosummary::
+    :nosignatures:
+    :toctree: generated
+
+    cov
+    nanmax
+    nanmin
+    nansum
+```