tinygrad logo

Tinygrad

tinygrad is an open-source, simple neural network framework for developing and running machine-learning models.
Category: Developer Tooling

Description

tinygrad is a neural network framework maintained by tiny corp. It provides a PyTorch-like API, automatic differentiation with full forward and backward passes, ONNX loading, training, inference, lazy tensors, operation fusion, and custom kernel compilation across hardware backends.