Jacky Lee
|
9fd41632c6
|
Import get_parameters from tinygrad.nn (#559)
* get_parameter is in optim
* Update all imports for get_parameters
* Clean up
* use optim.get_paramters
|
2023-02-17 15:22:26 -08:00 |
|
Jacky Lee
|
e172f0087a
|
BatchNorm2D -> BatchNorm2d (#558)
* BatchNorm2D -> BatchNorm2d
* Fix typo
|
2023-02-16 12:31:49 -08:00 |
|
George Hotz
|
fff1f046b0
|
Simple version of the new GPU backend (#458)
* newgpu
* more to delete
* hmm, tests pass with constant folding
* fix lint/type
* fix constant folding
* comment and rerun tests
* lazy touchups
* fix graph_batchnorm test
* smaller transformer to fix OOM
* Revert "smaller transformer to fix OOM"
This reverts commit a44ef8edc2.
* no func cache
* introspect
* touchups
* CLASTKernel
* ugh, it was lru_cache
* codegen
* spacing
* old gpu still in opencl
* typing fix
|
2023-01-10 19:16:02 -08:00 |
|
George Hotz
|
7f15779942
|
t.assign in optim
|
2022-08-20 14:04:33 -07:00 |
|
George Hotz
|
1eb12dafbc
|
reduce axis at the end
|
2022-08-20 07:40:56 -07:00 |
|
George Hotz
|
f76d41812b
|
prune graph
|
2022-07-17 15:38:43 -07:00 |
|
George Hotz
|
f9a8412b68
|
make contiguous ops yellow
|
2022-07-02 17:54:04 -07:00 |
|
George Hotz
|
207b9e1df3
|
padding is now a param to conv2d
|
2022-07-02 17:11:12 -07:00 |
|