jax-js
System
Light
Dark
Search…
Preparing search index...
@jax-js/optax
adam
Function adam
adam
(
learningRate
:
number
|
(
(
count
:
number
)
=>
number
)
,
opts
?:
ScaleByAdamOptions
,
)
:
GradientTransformation
The Adam optimizer.
Parameters
learningRate
:
number
|
(
(
count
:
number
)
=>
number
)
Optional
opts
:
ScaleByAdamOptions
Returns
GradientTransformation
jax-js
Loading...
The Adam optimizer.