docs update

2025-10-17 19:14:55 +00:00
parent 3ee92b5d61
commit d8f3a8a834
371 changed files with 32812 additions and 6257 deletions
@@ -0,0 +1,6 @@
+mlx.nn.AvgPool1d
+================
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: AvgPool1d
@@ -0,0 +1,6 @@
+mlx.nn.AvgPool2d
+================
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: AvgPool2d
@@ -0,0 +1,6 @@
+mlx.nn.MaxPool1d
+================
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: MaxPool1d
@@ -0,0 +1,6 @@
+mlx.nn.MaxPool2d
+================
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: MaxPool2d
@@ -10,6 +10,8 @@ Layers
   :template: nn-module-template.rst

   ALiBi
+   AvgPool1d
+   AvgPool2d
   BatchNorm
   Conv1d
   Conv2d
@@ -22,6 +24,8 @@ Layers
   InstanceNorm
   LayerNorm
   Linear
+   MaxPool1d
+   MaxPool2d
   Mish
   MultiHeadAttention
   PReLU