tinygrad

We write and maintain tinygrad, the fastest growing neural network framework

It's extremely simple, and breaks down the most complex networks into 3 OpTypes

ElementwiseOps are UnaryOps, BinaryOps, and TernaryOps.
They operate on 1-3 tensors and run elementwise.
example: SQRT, LOG2, ADD, MUL, WHERE, etc...

ReduceOps operate on one tensor and return a smaller tensor.
example: SUM, MAX

MovementOps are virtual ops that operate on one tensor and move the data around copy-free.
example: RESHAPE, PERMUTE, EXPAND, etc...

But how...where are your CONVs and MATMULs? Read the code to solve this mystery.


Work at tiny corp

We are now funded and hiring full time software engineers. Very talented interns okay.

See our bounty page to judge if you might be a good fit. Bounties pay you while judging that fit.

We are also hiring for operations and hardware, but if you haven't contributed to tinygrad your application won't be considered.

tinybox (now shipping)

Meet the new tinybox. Choose your memory, storage, and GPUs. Configure yours →

We sell a computer called the tinybox. It comes in red, green, pro, and soon, exa.

red v2green v2 blackwellexabox
FP16 (FP32 acc) FLOPS778 TFLOPS3086 TFLOPS~1 EXAFLOP
GPU Model4x 9070XT4x RTX PRO 6000 Blackwell720x RDNA5 AT0 XL
GPU RAM64 GB384 GB25,920 GB
GPU RAM bandwidth2560 GB/s7168 GB/s1244 TB/s
GPU link bandwidthfull fabric PCIe 4.0 x16full fabric PCIe 5.0 x16full fabric 400 GbE
CPU32 core AMD EPYC32 core AMD GENOA120x 32 core AMD GENOA
System RAM128 GB192 GB23,040 GB
System RAM bandwidth204.8 GB/s460.8 GB/s55.2 TB/s
Disk size2 TB fast NVMe4 TB raid + 1 TB boot480 TB raid
Disk read bandwidth7.3 GB/s59.3 GB/s7.1 TB/s
Networking2x 1GbE + OCP3.02x 10GbE + OCP3.0 PCIe53.2 TB/s scale out
Noise< 50 dB, 31 low speed fans65 db @ 10 meters
Power Supplyone 1600W, 100V~240V2x 1600W, 100V~240V600 kW, 200V~240V
BMCAST2500AST2600custom
Operating SystemUbuntu 24.04
Dimensions12U, 16.25" deep, 60-90 lbs20x8x8.5 ft, 20,000 lbs
Rack?Freestanding or rack mountconcrete slab
Driver QualityGoodGreatfunctions as single GPU
SHIPPINGORDER NOWORDER NOWPREORDER FOR 2027

more products on the shop. for updates on products and inventory, sign up for the mailing list