tinygrad is a compact end-to-end deep learning stack designed to sit somewhere between PyTorch and micrograd. It combines a tensor library, automatic differentiation, compiler, JIT execution, and neural network facilities in a deliberately small and hackable codebase.
Its execution engine supports a wide range of hardware backends while exposing much of the compiler and intermediate representation to developers.
This is free and open source software.
Key Features
- Tensor library with automatic differentiation.
- Intermediate representation and compiler for lowering computational kernels.
- Lazy execution enables operations to be fused into efficient kernels.
- JIT compilation and graph execution.
- Includes neural network layers and optimisers.
- Supports familiar PyTorch-style training loops.
- Provides datasets and other components needed for model training.
- Runs on CPUs and numerous hardware accelerators.
- Supports CUDA, AMD, Metal, OpenCL, Qualcomm, and WebGPU backends.
- Compiler and intermediate representation are designed to remain readable and hackable.
- Small codebase intended to minimise complexity.
Website: https://github.com/tinygrad/tinygrad
Support:
Developer: tiny corp
License: MIT License
tinygrad is written in Python and C. Learn Python with our recommended free books and free tutorials. Learn C with our recommended free books and free tutorials.
Related Software
| Deep Learning with Python | |
|---|---|
| TensorFlow | A very popular Deep Learning framework |
| PyTorch | Tensors and Dynamic neural networks in Python |
| Keras | High-level neural networks API |
| fastai | Simplifies training fast and accurate neural nets using modern best practices |
| PyTensor | Library for fast numerical computation |
| Elephas | Distributed deep learning with Keras and Spark |
| Chainer | Powerful, flexible, and intuitive framework for neural networks |
| Caffe | Convolutional Architecture for Fast Feature Embedding |
| TFlearn | Deep learning library featuring a higher-level API for TensorFlow |
| MXNet | Flexible and efficient library |
| CNTK | Distributed deep learning |
| Neupy | Python library for Artificial Neural Networks and Deep Learning |
Read our verdict in the software roundup.
 Explore our comprehensive directory of recommended free and open source software. Our carefully curated collection spans every major software category.This directory is part of our ongoing series of informative articles for Linux enthusiasts. It features hundreds of detailed reviews, along with open source alternatives to proprietary solutions from major corporations such as Google, Microsoft, Apple, Adobe, IBM, Cisco, Oracle, and Autodesk. You’ll also find interesting projects to try, hardware coverage, free programming books and tutorials, and much more. Discovered a useful open source Linux program that we haven’t covered yet? Let us know by completing this form. |
Please read our Comment Policy before commenting.