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