Thanks to visit codestin.com
Credit goes to github.com

Skip to content

ostadgeorge/regrad

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 

Repository files navigation

regrad

backtrack in rust

Usage

Add to Cargo.toml

cargo add regrad

Example

use regrad::Value;

fn main() {
    let v1 = Value::from(1.2);
    let v2 = Value::from(3.4);
    let v3 = &(&v1 * &v1) * &v2;

    dbg!(v3.data());
    assert_eq!(v3.data(), 4.896);

    v3.backward();
    dbg!(v1.gradient());
    dbg!(v2.gradient());
    dbg!(v3.gradient());

    assert_eq!(v1.gradient(), 8.16);
    assert_eq!(v2.gradient(), 1.44);
    assert_eq!(v3.gradient(), 1.0);
}

TODO

  • Add more operations
  • Add Tensor support
  • Add more tests
  • Add more examples
  • Add more documentation
  • Add GPU support
  • Crates.io publish

About

micrograd in rust

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages