github链接:https://:github.com/jinbooooom/OriginDLOriginDL: 完全从零开始构建的深度学习框架OriginDL 是一个完全从零开始构建的深度学习框架,采用 C++ 实现。OriginDL 不仅实现了高层的自动微分系统,更从最底层的矩阵运算开始,使用 CUDA 手写了 GPU 加速的矩阵计算核心,并在此基础上构建了完整的自动求导引擎和神经网络模块。目前已在框架上实现了线性回归与 MNIST 手写数字识别的训练示例,以及基于 PNNX 的 YOLOv5 目标检测推理,用于验证自动求导与推理链路。OriginDL 提供了类似 PyTorch 的 A...