pub trait Optimizer: Debug { // Required method fn step(&mut self, weights: &mut Tensor, gradients: &Tensor); }