rebase

2025-09-19 02:38:09 +08:00 · 2024-11-22 20:24:16 +00:00
parent a84697024f
commit 379b7b4027
905 changed files with 30035 additions and 16934 deletions
--- a/docs/build/html/_sources/python/_autosummary/mlx.core.fast.affine_quantize.rst
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.fast.affine_quantize.rst
@@ -1,6 +0,0 @@
-mlx.core.fast.affine\_quantize
-==============================
-
-.. currentmodule:: mlx.core.fast
-
-.. autofunction:: affine_quantize
--- a/docs/build/html/_sources/python/fast.rst
+++ b/docs/build/html/_sources/python/fast.rst
@@ -12,5 +12,4 @@ Fast
  layer_norm
  rope
  scaled_dot_product_attention
-  affine_quantize
  metal_kernel
--- a/docs/build/html/_sources/python/nn/_autosummary/mlx.nn.AvgPool3d.rst
+++ b/docs/build/html/_sources/python/nn/_autosummary/mlx.nn.AvgPool3d.rst
@@ -0,0 +1,16 @@
+mlx.nn.AvgPool3d
+================
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: AvgPool3d
+
+   
+
+   
+   .. rubric:: Methods
+
+   .. autosummary::
+   
+   
+   
--- a/docs/build/html/_sources/python/nn/_autosummary/mlx.nn.MaxPool3d.rst
+++ b/docs/build/html/_sources/python/nn/_autosummary/mlx.nn.MaxPool3d.rst
@@ -0,0 +1,16 @@
+mlx.nn.MaxPool3d
+================
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: MaxPool3d
+
+   
+
+   
+   .. rubric:: Methods
+
+   .. autosummary::
+   
+   
+   
--- a/docs/build/html/_sources/python/nn/layers.rst
+++ b/docs/build/html/_sources/python/nn/layers.rst
@@ -12,6 +12,7 @@ Layers
   ALiBi
   AvgPool1d
   AvgPool2d
+   AvgPool3d
   BatchNorm
   CELU
   Conv1d
@@ -41,6 +42,7 @@ Layers
   LSTM
   MaxPool1d
   MaxPool2d
+   MaxPool3d
   Mish
   MultiHeadAttention
   PReLU