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

   elu
   gelu
   gelu_approx
   gelu_fast_approx
   glu
   hardswish
   leaky_relu
   log_sigmoid
   log_softmax
   mish
   prelu
   relu
   relu6
   selu
   sigmoid
   silu
   softmax
   softplus
   softshrink
   step
   tanh
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`

Fix compile with non standard types (#745) * refactor tree utils * fix compile + tree code refactor * Add an extra test * add a few missing activations to docs * hash structure * Encode the full argument structure --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com> 2024-02-27 11:28:53 +08:00			`elu`
Docs updates (#198) Reorganize NN docs + a few other tidbits. 2023-12-18 05:20:55 +08:00			`gelu`
			`gelu_approx`
			`gelu_fast_approx`
Fix compile with non standard types (#745) * refactor tree utils * fix compile + tree code refactor * Add an extra test * add a few missing activations to docs * hash structure * Encode the full argument structure --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com> 2024-02-27 11:28:53 +08:00			`glu`
			`hardswish`
			`leaky_relu`
			`log_sigmoid`
			`log_softmax`
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`
Fix compile with non standard types (#745) * refactor tree utils * fix compile + tree code refactor * Add an extra test * add a few missing activations to docs * hash structure * Encode the full argument structure --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com> 2024-02-27 11:28:53 +08:00			`relu6`
Sort some APIs docs by names (a-z) (#472) 2024-01-17 11:37:50 +08:00			`selu`
Fix compile with non standard types (#745) * refactor tree utils * fix compile + tree code refactor * Add an extra test * add a few missing activations to docs * hash structure * Encode the full argument structure --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com> 2024-02-27 11:28:53 +08:00			`sigmoid`
Docs updates (#198) Reorganize NN docs + a few other tidbits. 2023-12-18 05:20:55 +08:00			`silu`
Fix compile with non standard types (#745) * refactor tree utils * fix compile + tree code refactor * Add an extra test * add a few missing activations to docs * hash structure * Encode the full argument structure --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com> 2024-02-27 11:28:53 +08:00			`softmax`
			`softplus`
			`softshrink`
Docs updates (#198) Reorganize NN docs + a few other tidbits. 2023-12-18 05:20:55 +08:00			`step`
Fix compile with non standard types (#745) * refactor tree utils * fix compile + tree code refactor * Add an extra test * add a few missing activations to docs * hash structure * Encode the full argument structure --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com> 2024-02-27 11:28:53 +08:00			`tanh`