George Hotz 7276f8d6bf improve constant folding, detach before moving tensor		2022-07-02 15:29:40 -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	comments	2022-06-26 17:20:25 -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