This commit is contained in:
Awni Hannun
2023-12-17 13:23:03 -08:00
committed by CircleCI Docs
parent 65b0bdffe1
commit 9d7133097f
319 changed files with 39952 additions and 21579 deletions

View File

@@ -0,0 +1,8 @@
mlx.nn.Conv1d
=============
.. currentmodule:: mlx.nn
.. autoclass:: Conv1d

View File

@@ -0,0 +1,8 @@
mlx.nn.Conv2d
=============
.. currentmodule:: mlx.nn
.. autoclass:: Conv2d

View File

@@ -0,0 +1,8 @@
mlx.nn.Embedding
================
.. currentmodule:: mlx.nn
.. autoclass:: Embedding

View File

@@ -0,0 +1,8 @@
mlx.nn.GELU
===========
.. currentmodule:: mlx.nn
.. autoclass:: GELU

View File

@@ -0,0 +1,8 @@
mlx.nn.GroupNorm
================
.. currentmodule:: mlx.nn
.. autoclass:: GroupNorm

View File

@@ -0,0 +1,8 @@
mlx.nn.LayerNorm
================
.. currentmodule:: mlx.nn
.. autoclass:: LayerNorm

View File

@@ -0,0 +1,8 @@
mlx.nn.Linear
=============
.. currentmodule:: mlx.nn
.. autoclass:: Linear

View File

@@ -0,0 +1,8 @@
mlx.nn.Mish
===========
.. currentmodule:: mlx.nn
.. autoclass:: Mish

View File

@@ -0,0 +1,8 @@
mlx.nn.MultiHeadAttention
=========================
.. currentmodule:: mlx.nn
.. autoclass:: MultiHeadAttention

View File

@@ -0,0 +1,8 @@
mlx.nn.PReLU
============
.. currentmodule:: mlx.nn
.. autoclass:: PReLU

View File

@@ -0,0 +1,8 @@
mlx.nn.RMSNorm
==============
.. currentmodule:: mlx.nn
.. autoclass:: RMSNorm

View File

@@ -0,0 +1,8 @@
mlx.nn.ReLU
===========
.. currentmodule:: mlx.nn
.. autoclass:: ReLU

View File

@@ -0,0 +1,8 @@
mlx.nn.RoPE
===========
.. currentmodule:: mlx.nn
.. autoclass:: RoPE

View File

@@ -0,0 +1,8 @@
mlx.nn.SELU
===========
.. currentmodule:: mlx.nn
.. autoclass:: SELU

View File

@@ -0,0 +1,8 @@
mlx.nn.Sequential
=================
.. currentmodule:: mlx.nn
.. autoclass:: Sequential

View File

@@ -0,0 +1,8 @@
mlx.nn.SiLU
===========
.. currentmodule:: mlx.nn
.. autoclass:: SiLU

View File

@@ -0,0 +1,8 @@
mlx.nn.Step
===========
.. currentmodule:: mlx.nn
.. autoclass:: Step