[package] name = "zenu-autograd" version = "0.1.2" edition = "2021" description = "A simple autograd library for learning purposes" license = "MIT" repository = "https://github.com/bokutotu/zenu" # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html [dependencies] zenu-matrix = { path = "../zenu-matrix", version = "0.1.2" } rand = "0.8.5" rand_distr = "0.4.3" lazy_static = "1.4.0" serde = { version = "1.0.197", features = ["derive"] } once_cell = "1.19.0" [features] nvidia = ["zenu-matrix/nvidia"] [dev-dependencies] criterion = "0.5.1" zenu-test = { path = "../zenu-test"} serde_json = "1.0.114" [[bench]] name = "conv2d" harness = false [profile.bench] debug = true