jax-js
    Preparing search index...

    Namespace nn

    Functions

    celu
    dotProductAttention
    elu
    gelu
    glu
    hardSigmoid
    hardSilu
    hardTanh
    identity
    leakyRelu
    logmeanexp
    logSigmoid
    logSoftmax
    logsumexp
    mish
    oneHot
    relu
    relu6
    selu
    sigmoid
    silu
    softmax
    softplus
    softSign
    sparsePlus
    sparseSigmoid
    squareplus
    standardize

    References

    hardSwish → hardSilu
    swish → silu
    Functions
    References