docs update

2025-10-17 19:14:55 +00:00
parent d8f3a8a834
commit 2bc2b00a93
378 changed files with 23040 additions and 2138 deletions
@@ -0,0 +1,6 @@
+mlx.nn.Upsample
+===============
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: Upsample
@@ -0,0 +1,6 @@
+mlx.nn.elu
+==========
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: elu
@@ -0,0 +1,6 @@
+mlx.nn.glu
+==========
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: glu
@@ -0,0 +1,6 @@
+mlx.nn.hardswish
+================
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: hardswish
@@ -0,0 +1,6 @@
+mlx.nn.leaky\_relu
+==================
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: leaky_relu
@@ -0,0 +1,6 @@
+mlx.nn.log\_sigmoid
+===================
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: log_sigmoid
@@ -0,0 +1,6 @@
+mlx.nn.log\_softmax
+===================
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: log_softmax
@@ -0,0 +1,6 @@
+mlx.nn.relu6
+============
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: relu6
@@ -0,0 +1,6 @@
+mlx.nn.sigmoid
+==============
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: sigmoid
@@ -0,0 +1,6 @@
+mlx.nn.softmax
+==============
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: softmax
@@ -0,0 +1,6 @@
+mlx.nn.softplus
+===============
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: softplus
@@ -0,0 +1,6 @@
+mlx.nn.tanh
+===========
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: tanh
@@ -12,13 +12,24 @@ simple functions.
   :toctree: _autosummary_functions
   :template: nn-module-template.rst

+   elu
   gelu
   gelu_approx
   gelu_fast_approx
+   glu
+   hardswish
+   leaky_relu
+   log_sigmoid
+   log_softmax
   mish
   prelu
   relu
+   relu6
   selu
-   softshrink
+   sigmoid
   silu
+   softmax
+   softplus
+   softshrink
   step
+   tanh
@@ -40,3 +40,4 @@ Layers
   Softshrink
   Step
   Transformer
+   Upsample