tinygrad/extra/models
wozeparrot d2c347fc74
faster gather for bert (#4526)
2024-05-10 22:28:48 -07:00
..
bert.py faster gather for bert (#4526) 2024-05-10 22:28:48 -07:00
convnext.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
efficientnet.py simple LoadOps.ASSIGN (#3745) 2024-03-14 20:44:34 -07:00
llama.py hotfix: revert llama change 2024-04-10 20:13:15 -07:00
mask_rcnn.py fix example that calls Tensor.__bool__ (#3650) 2024-03-07 16:59:26 -05:00
resnet.py fp16 resnet (without expand backwards sum in float, doesn't work) (#3816) 2024-03-28 01:25:37 -04:00
retinanet.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
rnnt.py create engine folder and move code (#3948) 2024-03-26 20:38:03 -07:00
transformer.py replace with tensor op (#3099) 2024-01-12 14:13:40 -05:00
unet3d.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
vit.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00