Optimizers

Java-friendly factory for optimizers.

Example usage from Java:

Optimizer opt = Optimizers.adam(0.001);
Optimizer sgd = Optimizers.sgd(0.01, 0.9);

Functions

fun adam(lr: Double = 0.001, beta1: Double = 0.9, beta2: Double = 0.999, epsilon: Double = 1.0E-8, weightDecay: Double = 0.0): Optimizer

Creates an Adam optimizer.

fun adamw(lr: Double = 0.001, beta1: Double = 0.9, beta2: Double = 0.999, epsilon: Double = 1.0E-8, weightDecay: Double = 0.01): Optimizer

Creates an AdamW optimizer (Adam with decoupled weight decay).

fun sgd(lr: Double, momentum: Double = 0.0, weightDecay: Double = 0.0): Optimizer

Creates an SGD optimizer with optional momentum.