pub fn adamw(beta1: f64, beta2: f64, wd: f64) -> AdamW
Creates the configuration for the AdamW optimizer.