极简主义的深度学习框架,旨在介于 PyTorch 和 micrograd 之间。其设计理念是简化框架,方便添加新加速器,支持推理和训练。Tinygrad 已支持多种加速器如 GPU、OpenCL、CUDA 等,能运行 LLaMA 和 Stable Diffusion。它提供了基本的自动求导和张量库,并有示例代码和文档指导使用
https://github.com/tinygrad/tinygrad
GitHub
GitHub - tinygrad/tinygrad: You like pytorch? You like micrograd? You love tinygrad! ❤️
You like pytorch? You like micrograd? You love tinygrad! ❤️ - GitHub - tinygrad/tinygrad: You like pytorch? You like micrograd? You love tinygrad! ❤️
via MJJ出征 - Telegram Channel