tinygrad/models
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