mlx/docs/src/python/nn/functions.rst

.. _nn_functions:

.. currentmodule:: mlx.nn

Functions
---------

Layers without parameters (e.g. activation functions) are also provided as
simple functions.

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

   gelu
   gelu_approx
   gelu_fast_approx
   mish
   prelu
   relu
   selu
   softshrink
   silu
   step
Docs updates (#198) Reorganize NN docs + a few other tidbits. 2023-12-18 05:20:55 +08:00			`.. _nn_functions:`

			`.. currentmodule:: mlx.nn`

			`Functions`
			`---------`

			`Layers without parameters (e.g. activation functions) are also provided as`
			`simple functions.`

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

			`gelu`
			`gelu_approx`
			`gelu_fast_approx`
Sort some APIs docs by names (a-z) (#472) 2024-01-17 11:37:50 +08:00			`mish`
Docs updates (#198) Reorganize NN docs + a few other tidbits. 2023-12-18 05:20:55 +08:00			`prelu`
Sort some APIs docs by names (a-z) (#472) 2024-01-17 11:37:50 +08:00			`relu`
			`selu`
Softshrink mapping + op (#552) * Added Softshrink mapping + op * formatting * docs + nits in docstring --------- Co-authored-by: Awni Hannun <awni@apple.com> 2024-01-31 04:56:28 +08:00			`softshrink`
Docs updates (#198) Reorganize NN docs + a few other tidbits. 2023-12-18 05:20:55 +08:00			`silu`
			`step`