George Hotz 90326dbdc3 resnet50 hand coded optimization (#1945) ...

* resnet50 hand coded opt

* hand optimize one kernel

* opt in both places to fix test

2023-09-29 09:34:51 -07:00

..

bert.py

[ready] Replacing os with pathlib (#1708)

2023-08-30 10:41:08 -07:00

convnext.py

LayerNorm2d for 2 lines

2023-03-20 16:58:43 -07:00

efficientnet.py

move state to nn/state (#1619)

2023-08-22 07:36:24 -07:00

mask_rcnn.py

move state to nn/state (#1619)

2023-08-22 07:36:24 -07:00

resnet.py

resnet50 hand coded optimization (#1945)

2023-09-29 09:34:51 -07:00

retinanet.py

Use generators instead of lists in anys and alls (#1111)

2023-07-03 16:06:06 -07:00

rnnt.py

[ready] Replacing os with pathlib (#1708)

2023-08-30 10:41:08 -07:00

transformer.py

.cpu().numpy() -> .numpy() (#1594)

2023-08-21 09:53:29 -07:00

unet3d.py

[ready] Replacing os with pathlib (#1708)

2023-08-30 10:41:08 -07:00

vit.py

good changes from llama branch (#671)

2023-03-09 20:51:22 -08:00