mlx/docs/build/html/_sources/python/nn/layers.rst

.. _layers:

.. currentmodule:: mlx.nn

Layers
------

.. autosummary::
   :toctree: _autosummary
   :template: nn-module-template.rst

   Embedding
   ReLU
   PReLU
   GELU
   SiLU
   Step
   SELU
   Mish
   Linear
   Conv1d
   Conv2d
   LayerNorm
   RMSNorm
   GroupNorm
   RoPE
   MultiHeadAttention
   Sequential
   QuantizedLinear
docs 2023-12-18 05:23:03 +08:00			`.. _layers:`

			`.. currentmodule:: mlx.nn`

			`Layers`
			`------`

			`.. autosummary::`
			`:toctree: _autosummary`
			`:template: nn-module-template.rst`

			`Embedding`
			`ReLU`
			`PReLU`
			`GELU`
			`SiLU`
			`Step`
			`SELU`
			`Mish`
			`Linear`
			`Conv1d`
			`Conv2d`
			`LayerNorm`
			`RMSNorm`
			`GroupNorm`
			`RoPE`
			`MultiHeadAttention`
			`Sequential`
docs 2023-12-22 14:13:41 +08:00			`QuantizedLinear`