zenu_optimizer/
lib.rs

1pub mod adam;
2pub mod adamw;
3pub mod sgd;
4
5use zenu_layer::Parameters;
6use zenu_matrix::{device::Device, num::Num};
7
8pub trait Optimizer<T: Num, D: Device, P: Parameters<T, D>> {
9    fn update(&self, parameters: &P);
10}