George Hotz 387ea41e99 increase speed of torch mnist: use gradient api (#9282)

2025-02-27 11:57:41 +08:00

..

.gitignore

torch backend works for ResNet-18 (#9200)

2025-02-22 22:16:23 +08:00

backend2.py

torch and numpy dtype interop [pr] (#9224)

2025-02-24 18:26:49 +08:00

backend.py

increase speed of torch mnist: use gradient api (#9282)

2025-02-27 11:57:41 +08:00

example.py

run some tests from torch [pr] (#9252)

2025-02-26 15:42:22 +08:00

test_compile.py

start torch.compile support (#9279)

2025-02-27 10:29:51 +08:00

test.py

(bounty) Make mnist training run with torch backend (#9233)

2025-02-27 11:32:25 +08:00

torch_tests.py

rig up torch's testing framework [pr] (#9254)

2025-02-26 18:46:22 +08:00

wrapped_tensor.cpp

full fix for as_strided in torch backend (#9257)

2025-02-26 22:34:05 +08:00