Struct GradientDescent 

pub struct GradientDescent { /* private fields */ }

SIMD-optimized gradient descent optimizer

Implementations

impl GradientDescent

pub fn new(learning_rate: f32) -> Self

Create a new gradient descent optimizer

pub fn with_momentum(self, momentum: f32) -> Self

Set momentum for the optimizer

pub fn with_weight_decay(self, weight_decay: f32) -> Self

Set weight decay (L2 regularization)

pub fn with_nesterov(self) -> Self

Enable Nesterov momentum

pub fn step( &self, params: &mut ArrayViewMut1<'_, f32>, gradient: &ArrayView1<'_, f32>, velocity: &mut ArrayViewMut1<'_, f32>, )

Perform a single optimization step