simian 0.1.1

A command-line tool for exploring and implementing Machine Learning algorithms in Rust.
Documentation 
1
2
3
4
5
6
7
8
9

use candle_core::{Result, Tensor};

pub mod sgd;

pub trait Optimizer {
  /// Given a list of mutable references to parameters and a list of their respective gradients,
  /// this function will update the parameters according to the optimizer's algorithm.
  fn step(&mut self, params: &mut [&mut Tensor], grads: &[Tensor]) -> Result<()>;
}