wyrm:Rust中的自动分化程序包

时间:2024-05-31 11:30:20
【文件属性】:

文件名称:wyrm:Rust中的自动分化程序包

文件大小:86KB

文件格式:ZIP

更新时间:2024-05-31 11:30:20

rust machine-learning artificial-intelligence neural-networks autodiff

飞龙 反向模式,按运行定义,低开销的自动分化库。 特征 通过任意的,按运行定义的计算图执行反向传播,强调对CPU上的稀疏小型模型进行低开销估计。 强调: 低开销。 内置对稀疏渐变的支持。 按运行定义。 琐碎的Hogwild风格并行化,可根据可用CPU内核的数量线性扩展。 快速开始 下面定义了单变量线性回归模型,然后通过它进行反向传播。 let slope = ParameterNode :: new ( random_matrix ( 1 , 1 )); let intercept = ParameterNode :: new ( random_matrix ( 1 , 1 )); let x = InputNode :: new ( random_matrix ( 1 , 1 )); let y = InputNode :: new ( random_matrix ( 1


【文件预览】:
wyrm-master
----.gitignore(33B)
----Cargo.toml(1KB)
----LICENSE(1KB)
----src()
--------numerics.rs(15KB)
--------lib.rs(39KB)
--------nn()
--------nodes.rs(73KB)
--------fast_approx.rs(4KB)
--------optim()
----benches()
--------benchmark.rs(7KB)
----readme.md(3KB)
----README.tpl(328B)
----.travis.yml(227B)

网友评论