#tensorflow #neural-network #framework #artificial #layer #bindings #gpu

Brains

基于TensorFlow的Rust人工神经网络框架

4个版本

0.2.0 2023年1月23日
0.1.2 2022年4月1日
0.1.1 2022年3月19日
0.1.0 2022年3月15日

#834 in 机器学习

GPL-3.0许可证

215KB
1.5K SLoC

Rust 1.5K SLoC // 0.2% comments Vim Script 239 SLoC

Brains

注意:此项目非常早期,起源于我自己的架构研究。欢迎贡献、意见和问题!1.X.X版本将视为可用于公共使用的版本。

基于TensorFlow-rs绑定构建的人工神经网络框架,可以创建类似keras的架构,同时也可以在低级Tensorflow中直接集成自定义层。包括本地检查点、推理、批量训练和迭代训练。请参阅单元测试以了解事物如何调用,直到创建文档。另外,如果您想卸载计算,请确保启用tensorflow的GPU标志。

目前所有输入和输出都表示为扁平化的1D rust Vecs。

一旦以下TODO完成,此项目将在crates上更新,并可以被视为一个初级框架(v1.X.X)

TODO

重构所有内容以使用特性和构建器模式,以获得更类似于Keras的顺序API用户界面,以及可以更轻松地构建和在其他框架(如libtorch)上实现的东西。

创建一个dev分支并将其拉入Master作为发布分支

文档。文档字符串和项目概述页面,讨论Mod。

请告诉我您的看法: [email protected]

依赖项

~25MB
~546K SLoC