Skip to content

trizko/microrunn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

36 Commits
 
 
 
 
 
 
 
 

Repository files navigation

microrunn

Rust port of Andrej Karpathy's micrograd

Installation

cd microrunn/
cargo build
./target/debug/microrunn

Example usage

use microrunn::engine::Value;
use microrunn::nn::MLP;

fn main() {
    let x = vec![Value::new(0.0), Value::new(1.0)];
    let y = vec![Value::new(1.0)];
    let model: MLP = MLP::new(2, vec![3, 3, 1]);

    let out = model.call(&x);

    print!("actual: {:#?} | expected: {:#?}", out[0].data, y[0].data);
}

TODO

  • add loss functions
  • add optimizer
  • add model import/export
  • update example usage binary

About

Rust port of Andrej Karpathy's micrograd

Resources

Stars

Watchers

Forks

Languages