Docs / jax4j-core / com.nablatensor.jax4j.nn
final class
FunctionalAdam
Pure Adam updates: moments and the device step counter are explicit pytree inputs and outputs.
Constructors
FunctionalAdam(double rate)
FunctionalAdam(double rate, double beta1, double beta2, double epsilon)
Methods
State<P> initialize(P parameters)
Update<P> update(P parameters, P gradients, State<P> state)