Functions

.. _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 celu gelu gelu_approx gelu_fast_approx glu hard_shrink hard_tanh hardswish leaky_relu log_sigmoid log_softmax mish prelu relu relu2 relu6 selu sigmoid silu softmax softmin softplus softshrink step tanh