mlx/docs/src/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
   BatchNorm
   LayerNorm
   RMSNorm
   GroupNorm
   RoPE
   MultiHeadAttention
   Sequential
   QuantizedLinear
   Dropout
   Dropout2d
Docs updates (#198) Reorganize NN docs + a few other tidbits. 2023-12-18 05:20:55 +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`
implement-batch-norm-layer (#217) - Add batch normalization layer --------- Co-authored-by: Robert McCraith <mccraithrobert@gmail.com> Co-authored-by: Awni Hannun <awni@apple.com> 2023-12-25 23:32:53 +08:00			`BatchNorm`
Docs updates (#198) Reorganize NN docs + a few other tidbits. 2023-12-18 05:20:55 +08:00			`LayerNorm`
			`RMSNorm`
			`GroupNorm`
			`RoPE`
			`MultiHeadAttention`
			`Sequential`
Adds C++ and nn quantization utilities (#230) * Add C++ de-/quantize ops * Add quantize functions to the docs and tests * Add a QuantizedLinear module 2023-12-21 06:17:38 +08:00			`QuantizedLinear`
Add dropout2d (#250) 2023-12-23 00:02:29 +08:00			`Dropout`
			`Dropout2d`