tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-09-02 09:26:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
90fb09b55c5960f275e741b1049b24f40fce1b3c
tinygrad/extra/models
T
History
chenyuandGitHub c0f76ed4ea transformer kvcache and mask have same dtype as input (#2771) 
* transformer kvcache and mask have same dtype as input

* don't use `=0` in cstyle ternary where

* (bool)

* where float16 test
2023-12-14 22:41:51 -05:00
..
bert.py
move to new cached fetch (#2493)
2023-11-28 17:36:55 -08:00
convnext.py
move to new cached fetch (#2493)
2023-11-28 17:36:55 -08:00
efficientnet.py
add name support to fetch (#2407)
2023-11-23 14:16:17 -08:00
llama.py
transformer kvcache and mask have same dtype as input (#2771)
2023-12-14 22:41:51 -05:00
mask_rcnn.py
move to new cached fetch (#2493)
2023-11-28 17:36:55 -08:00
resnet.py
move to new cached fetch (#2493)
2023-11-28 17:36:55 -08:00
retinanet.py
move to new cached fetch (#2493)
2023-11-28 17:36:55 -08:00
rnnt.py
move to new cached fetch (#2493)
2023-11-28 17:36:55 -08:00
transformer.py
move things, clean up extra (#2292)
2023-11-13 20:18:40 -08: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