https://github.com/tinygrad/teenygrad 简易的深度学习框架,基于CPU上的numpy。在上层封装了Tensor的各种操作,自动微分。是一个用来学习自动微分引擎的比较不错的小项目。代码量也非常小。 同时有一个扩展版本的项目tinygrad,在此之上支持了多种后端,可以看作是pytorch的缩小版 个人感觉需要关注的点主要是三个: 数据的表示(Tensor) Autograd 常见操作的实现 Tensor Tensor是一个N维的数组,在此之上,支持: 多种常见的计算函数 自动微分 …