| .. _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 | |