update readme

This commit is contained in:
George Hotz
2020-10-18 13:08:14 -07:00
parent 83417d4b4c
commit 54eafe6c12
2 changed files with 32 additions and 24 deletions

View File

@@ -2,7 +2,9 @@
For something in between a grad and a karpathy/micrograd
The Tensor class is a wrapper around a numpy array
This may not be the best deep learning framework, but it is a deep learning framework.
The Tensor class is a wrapper around a numpy array, except it does Tensor things.
### Example
@@ -33,3 +35,9 @@ print(x.grad) # dz/dx
print(y.grad) # dz/dy
```
### TODO (to make real neural network library)
* Implement convolutions
* Implement Adam optimizer