George Hotz a949de873b reduce 2.0 (#469) ...

* reduce 2.0

* works

* hacks

* DEBUG=3 for shapes

* fix types

* 0s weren't being folded

* cleaner

* last_reduce is no longer needed

* comments and cleanup

2023-01-23 15:11:13 -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

Simple version of the new GPU backend (#458)

2023-01-10 19:16:02 -08:00

onnx.py

Typing (#468)

2023-01-21 09:09:22 -08:00

thneed.py

reduce 2.0 (#469)

2023-01-23 15:11:13 -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