AIAny
AI Infra2020
Icon for item

tinygrad

Sits between PyTorch and micrograd: eager tensors with autograd plus a small, fully hackable compiler that fuses operations into kernels. Adding a new accelerator backend takes about 25 low-level ops, so it runs on CUDA, Metal, AMD, and WebGPU.

Introduction

Every serious deep learning framework eventually grows a compiler too large for any one person to hold in their head. tinygrad makes the opposite bet: keep the whole stack — from the eager tensor API down to the generated GPU kernel — small enough to read in an afternoon, and treat that readability as the product rather than a side effect. The leverage comes from one constraint: a new accelerator needs only about 25 low-level operations, so the framework, not the chip vendor, owns the path to new hardware.

What Sets It Apart
  • One visible IR: the eager and compiled paths don't diverge the way they do in PyTorch — everything lowers through a single lazy IR you can inspect and patch, so debugging a fused kernel means reading tinygrad, not a closed vendor runtime.
  • ~25 ops per backend: bringing up CUDA, Metal, AMD, OpenCL, WebGPU, or plain CPU means implementing a tiny operation set, turning new-hardware support into something one motivated contributor can do.
  • Fusion without leaving eager: operations build a lazy graph that gets fused and scheduled, with BEAM search over kernel variants, so you get compiler-level speedups while the API still feels immediate.
  • Funded by hardware: the tiny corp sells the tinybox around it, so the project is sustained by shipping machines rather than cloud credits, and its incentives point at running well on commodity AMD and NVIDIA silicon.
Who It's For

Great fit if you want to actually see how a tensor becomes a GPU kernel, are bringing up an unusual accelerator, or value a codebase small enough to audit end to end. Look elsewhere if you need a mature ecosystem today: the API still shifts between releases, pretrained-model and tooling coverage trails PyTorch, and on well-trodden NVIDIA paths the incumbents can still win on raw performance.

Information

  • Websitegithub.com
  • Organizationsthe tiny corp
  • Authorstiny corp
  • Published date2020/10/18

Categories

More Items

GitHub
AI Infra2025

Defines a vendor-neutral JSON/YAML semantic model specification and tooling to exchange metrics, dimensions, lineage and other business semantics across analytics, AI and BI platforms; includes a core spec, validators, converters (dbt, GoodData, Salesforce) and example models.

GitHub
AI Train2025

An asynchronous, high-throughput framework for large-scale reinforcement learning and agentic training that scales to 1T+ MoE models and 1000+ GPUs, with native verifiers integration, end-to-end SFT/RL/evals, and Slurm/Kubernetes deployment; requires NVIDIA GPUs.

GitHub

Runs a self-hosted meeting bot and transcription API that joins Google Meet, Teams and Zoom and streams speaker-attributed transcripts in real time. Compiles meetings into a git-backed Markdown workspace and runs sandboxed agents on your infrastructure; Apache-2.0 and air-gap capable.