Skip to main content

Browse the archive

Revision 7b182ec8791c97ba61bbe186e956e7bdd340a0f0 authored by Brian Patton on 11 March 2024, 16:05:25 UTC, committed by jax authors on 11 March 2024, 16:56:55 UTC 
Modify test to detect erroneous sorting of negative int32 as larger than positive int32.
 
PiperOrigin-RevId: 614684199
1 parent 71ec6e3
Raw File
jax.nn.rst 

``jax.nn`` module
=================

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

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

.. autosummary::
  :toctree: _autosummary

    softmax
    log_softmax
    logsumexp
    standardize
    one_hot
back to top