pub trait VariationalOptimizer: Send + Sync {
// Required methods
fn update_parameters(
&mut self,
parameters: Vec<f64>,
gradients: Vec<f64>,
) -> DeviceResult<Vec<f64>>;
fn reset(&mut self);
}Expand description
Trait for variational optimizers