tinygrad/extra
George Hotz 81b73f97a3
Optiimzation (#355)
* constant folding into kernels

* that opt worth it?

* fix mypy

* ast one kernel

* save 2 lines in conv kernel

* debug print kernel count

* cl debugging

* early realize inputs

* refactor Device
2022-07-04 08:58:57 -07:00
..
augment.py minor optimizations & cleaning (#257) 2021-06-02 09:57:15 -07:00
gradcheck.py move gradcheck to extra, clean up unbroadcast 2020-11-16 08:03:31 -08:00
onnx.py Optiimzation (#355) 2022-07-04 08:58:57 -07:00
training.py improve constant folding, detach before moving tensor 2022-07-02 15:29:40 -07:00
utils.py don't track_running_stats, parameters must require_grad 2022-07-02 14:38:45 -07:00