Revision 66b182b735503db3a7bbc956b2fc1d30d897505b authored by Matthew Johnson on 06 December 2019, 15:44:12 UTC, committed by Matthew Johnson on 06 December 2019, 19:12:59 UTC
1 parent 1da950d
Raw File
jax.nn.rst

jax.nn package
=================

.. currentmodule:: jax.nn

.. toctree::
    :maxdepth: 1

    jax.nn.initializers

.. automodule:: jax.nn


Activation functions
------------------------

.. autosummary::
  :toctree: _autosummary

    relu
    sigmoid
    softplus
    soft_sign
    swish
    log_sigmoid
    leaky_relu
    hard_tanh
    elu
    celu
    selu
    gelu
    glu

Other functions
---------------

.. autosummary::
  :toctree: _autosummary

    softmax
    log_softmax
    normalize
back to top