George Hotz bb288e6938 safe_numpy and warning for broken matmul

2022-10-20 15:40:22 -07: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

fix stupid OPENCL=1 OOM

2022-09-06 14:29:23 -07:00

onnx.py

safe_numpy and warning for broken matmul

2022-10-20 15:40:22 -07:00

thneed.py

add assert to catch issue in attention

2022-10-20 15:13:00 -07: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