docs update

This commit is contained in:
Awni Hannun
2024-02-17 13:25:37 -08:00
committed by CircleCI Docs
parent e492638dff
commit 2c714de62a
371 changed files with 32812 additions and 6257 deletions

View File

@@ -0,0 +1,6 @@
mlx.nn.AvgPool1d
================
.. currentmodule:: mlx.nn
.. autoclass:: AvgPool1d

View File

@@ -0,0 +1,6 @@
mlx.nn.AvgPool2d
================
.. currentmodule:: mlx.nn
.. autoclass:: AvgPool2d

View File

@@ -0,0 +1,6 @@
mlx.nn.MaxPool1d
================
.. currentmodule:: mlx.nn
.. autoclass:: MaxPool1d

View File

@@ -0,0 +1,6 @@
mlx.nn.MaxPool2d
================
.. currentmodule:: mlx.nn
.. autoclass:: MaxPool2d

View File

@@ -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