rocm_jax/docs/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
add docs for jax.nn 2019-08-29 17:51:15 -07:00
			`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`
Document functions in jax.nn. (#1795) 2019-12-02 14:21:10 -05:00			`glu`
add docs for jax.nn 2019-08-29 17:51:15 -07:00
			`Other functions`
			`---------------`

			`.. autosummary::`
			`:toctree: _autosummary`

			`softmax`
			`log_softmax`
			`normalize`