jax-js
    Preparing search index...

    Function sgd

    • Stochastic gradient descent.

      Parameters

      • learningRate: number | ((count: number) => number)

      Returns GradientTransformation