#machine-learning #automatic-differentiation #auto-diff #neural-network #math

nightly gtensor

使用张量进行计算图的反向自动微分,适用于机器学习及其他

4个版本 (1个稳定版)

1.0.0 2023年11月27日
0.5.4 2022年12月26日
0.2.1 2023年11月27日
0.1.1 2023年11月27日

#407 in 机器学习

MIT/Apache

79KB
2K SLoC

图张量

一个用于机器学习以及更多操作的计算图上张量运算的逆模式自动微分库。

目标

gTensor的目标是创建一个以性能、灵活性和文档为中心的通用机器学习框架。我们希望涵盖深度、卷积和循环神经网络,以及无监督算法如KNN和聚类,以及强化算法如深度Q学习。

文档

在/docs/文件夹中提供了大量的文档。

示例

目前gT提供了classification示例,展示了如何加载数据集、构建、训练和测试神经网络,以及将网络保存到磁盘上。

贡献

gTensor处于积极早期开发阶段。预计会有频繁的破坏性更改。如果您发现gT缺少重要的功能,请随时创建一个pull request。

依赖关系

~10MB
~183K SLoC