github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-08 14:43:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

more readme

Browse Source
This commit is contained in:
George Hotz
2020-11-02 08:33:48 -08:00
parent 1e6bbdf4f8
commit 940e14c6ca
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
README.md
View File

@@ -12,8 +12,6 @@ This may not be the best deep learning framework, but it is a deep learning fram
The Tensor class is a wrapper around a numpy array, except it does Tensor things.
Update: tinygrad also somewhat supports GPUs through pyopencl. Our end goal is to train an ImageNet EfficientNet-B0.
### Installation
```bash
@@ -78,10 +76,19 @@ loss.backward()
optim.step()
```
### GPU Support?!
tinygrad supports GPUs through PyOpenCL. Not all ops are supported yet.
```python
from tinygrad.tensor import Tensor
(Tensor.ones(4,4).cuda() + Tensor.ones(4,4).cuda()).cpu()
```
### ImageNet inference (on the micrograd puppy)
```bash
python3 examples/examples/efficientnet.py
python3 examples/efficientnet.py
```
### The promise of small