add losses to the docs, fix black failur (#92)

2025-09-26 15:58:14 +08:00 · 2023-12-09 06:06:52 -08:00
parent 430bfb4944
commit 2520dbcf0a
2 changed files with 13 additions and 6 deletions
--- a/docs/src/python/nn.rst
+++ b/docs/src/python/nn.rst
@@ -170,3 +170,13 @@ simple functions.
   gelu_fast_approx
   relu
   silu
+
+Loss Functions
+--------------
+
+.. autosummary::
+   :toctree: _autosummary_functions
+   :template: nn-module-template.rst
+
+   losses.cross_entropy
+   losses.l1_loss