Revision 6c05add8ae69fb25e09b3bb537ea90a2a3033b95 authored by yashkatariya on 15 November 2022, 20:59:00 UTC, committed by yashkatariya on 15 November 2022, 20:59:00 UTC
1 parent 233bf21
jax.nn.rst
jax.nn package
=================
.. currentmodule:: jax.nn
.. toctree::
:maxdepth: 1
jax.nn.initializers
.. automodule:: jax.nn
Activation functions
------------------------
.. autosummary::
:toctree: _autosummary
relu
relu6
sigmoid
softplus
soft_sign
silu
swish
log_sigmoid
leaky_relu
hard_sigmoid
hard_silu
hard_swish
hard_tanh
elu
celu
selu
gelu
glu
Other functions
---------------
.. autosummary::
:toctree: _autosummary
softmax
log_softmax
logsumexp
standardize
one_hot
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...