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

Measures generative AI inference performance with token-level metrics (TTFT, inter-token latency), latency, and throughput under realistic traffic patterns. Provides a multiprocess engine, real-time TUI dashboard, extensible plugins, and integrations for telemetry and result uploads, aimed at inference benchmarking and capacity planning.

GitHub
AI Train2019

Train and experiment with multi-billion to trillion-parameter transformer models on large GPU clusters using GPU-optimized building blocks and reference training scripts; offers advanced parallelism and mixed-precision support for research teams and ML engineers.

GitHub

Indexes full text of visited web pages and local files on a self‑hosted server so you can search your personal knowledge from a web UI, terminal, CLI, or an AI assistant. Runs without mandatory telemetry, offers a browser extension for automatic capture, and supports optional semantic search via a configurable embeddings endpoint.