rai_opt/optimizers/
mod.rs

1
2
3
4
5
6
7
8
9
10
use rai_core::Tensor;
use std::collections::HashMap;

pub trait Optimizer {
    /// return new value for the parameters
    fn step(&mut self, grads: &HashMap<usize, Tensor>) -> HashMap<usize, Tensor>;
}

mod sdg;
pub use sdg::SDG;