【文件属性】：
文件名称：tinygrad:你喜欢pytorch吗？ 你喜欢研究生吗？ 你爱tinygrad！ :red_heart:
文件大小：148KB
文件格式：ZIP
更新时间：2021-03-16 02:34:27
Python
对于介于和之间的东西
这可能不是最佳的深度学习框架，但它是一个深度学习框架。
由于其极其简单，因此它旨在成为添加新加速器的最简单框架，同时支持推理和训练。 支持简单的基本操作，您将获得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