tinygrad:你喜欢pytorch吗? 你喜欢研究生吗? 你爱tinygrad! :red_heart: 源码

对于介于和之间的东西 这可能不是最佳的深度学习框架,但它是一个深度学习框架。 由于其极其简单,因此它旨在成为添加新加速器的最简单框架,同时支持推理和训练。 支持简单的基本操作,您将获得SOTAextra/efficientnet.py和extra/transformer.py模型。 我们正在努力支持Apple Neural Engine。 最终,我们将为tinygrad构建自定义硬件,并且速度将非常快。 现在,它很慢。 安装 pip3 install git+https://github.com/geohot/tinygrad.git --upgrade 例子 from tinygrad . tensor import Tensor x = Tensor . eye ( 3 ) y = Tensor ([[ 2.0 , 0 , - 2.0 ]]) z = y . matmul ( x