Content - 6fa496a9a6c11a09a62a6fc57ad04cf04cf5e8f1 - 0308561/docs/jax.nn.rst

visit type:

Tip revision: 639cda294c50b3a62c6cc7af0da7a367e5f876c3 authored by jax authors on 12 October 2020, 19:05:27 UTC
Merge pull request #4554 from skye:workspace

Tip revision: 639cda2

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
    normalize
    one_hot

Browse the archive

https://github.com/google/jax