George Hotz 4885fce56e shapetracker from newgpu (#456) ...

* shapetracker from newgpu

* touchup ops

* test

* testst

* thneed deletes unused inputs

* test

* bugfix

2023-01-09 12:40:01 -08:00

..

gemm

Simple chonker (#431)

2022-11-10 23:17:09 -08:00

augment.py

Reshape dataset from fetch_mnist (#390)

2022-09-24 21:16:29 -04:00

gradcheck.py

move gradcheck to extra, clean up unbroadcast

2020-11-16 08:03:31 -08:00

introspection.py

move ops.py into lazy.py (#402)

2022-10-25 13:58:03 -07:00

onnx.py

fix nonstatic weights

2022-10-20 17:04:14 -07:00

thneed.py

shapetracker from newgpu (#456)

2023-01-09 12:40:01 -08:00

training.py

prune graph

2022-07-17 15:38:43 -07:00

utils.py

move get_parameters to optim.py

2022-09-25 13:16:58 -04:00