輕量級深度學習框架tinygrad

tinygrad 是一個輕量級深度學習框架,它的目標是成為最容易添加新加速器的框架,同時支持推理和訓練。 tinygrad 支持簡單的基本操作,包含SOTA 視覺models/efficientnet.py和語言模型models/transformer.py。 tinygrad 正在努力支持accel/ 文件夾中的Apple Neural Engine 和Google TPU。最終將為tinygrad 構建定制硬件,以達到終極目標:極速推理/訓練框架。 (但現在它還很慢) 安裝git clone https://github.com/g…

#輕量級深度學習框架tinygrad

Leave a Reply

Your email address will not be published. Required fields are marked *