概要
tinygradはPyTorchとmicrogradの中間に位置する深層学習フレームワークで、tiny corpがメンテナンス。わずか約200のPythonソースファイルで、学習から推論までの完全なDLワークフローをサポート。
Python CUDA Metal WebGPU
コアコンポーネント
- Tensorライブラリ：PyTorchライクなEager API + 自動微分
- IRコンパイラ：カーネル融合・lowering、JIT + Graph実行
- LLM推論：GGUFフォーマット対応
ハードウェアバックエンド
NVIDIA CUDA/PTX, AMD ROCm/CDNA/RDNA3/RDNA4, Apple Metal, OpenCL, Qualcomm QCOM, WebGPU, LLVM IR, ONNX
深層学習 AIフレームワーク OSS