jax4j / NablaTensor

Docs / jax4j-core / com.nablatensor.jax4j.nn

final class

Adam

Adam with device-resident moments and fused elementwise updates.

View source ↗Full module reference on javadoc.io ↗

Constructors

Adam(float rate)

Methods

List<Tensor> update(List<Tensor> parameters, List<Tensor> gradients)

Returns new parameters. The caller owns parameters and gradients; this optimizer owns moments.

void close()