2 releases

new 0.1.1 Nov 14, 2024
0.1.0 Apr 1, 2024

#766 in Machine learning

Download history 10/week @ 2024-07-29 10/week @ 2024-09-23 3/week @ 2024-09-30 116/week @ 2024-11-11

116 downloads per month
Used in zenu

MIT license

1MB
23K SLoC

ZeNu Optimizer

ZeNu Optimizer is a collection of optimization algorithms for training neural networks. It provides various optimizers that can be used with the ZeNu deep learning library.

Features

  • Stochastic Gradient Descent (SGD) optimizer
  • Integration with ZeNu Autograd for gradient computation
  • Easy integration with ZeNu models and layers

Getting Started

To use ZeNu Optimizer in your Rust project, add the following to your Cargo.toml file:

[dependencies]
zenu-optimizer = "0.1.0"

Here's a simple example of using the SGD optimizer from ZeNu Optimizer:

use zenu_autograd::{creator::from_vec::from_vec, Variable};
use zenu_optimizer::sgd::SGD;

fn main() {
    let variable = from_vec(vec![1., 2., 3., 4., 5., 6.], [3, 2]);
    variable.set_grad(from_vec(vec![1., 2., 3., 4., 5., 6.], [3, 2]));

    let sgd = SGD::new(0.01);
    sgd.update(&[variable.clone()]);

    // The variable has been updated by the optimizer
    // Perform further computations with the updated variable
}

For more details and examples, please refer to the documentation.

License

ZeNu Optimizer is licensed under the MIT License.

Dependencies

~1.5–4MB
~72K SLoC