tinygrad/extra/models
Tobias Fischer f9e32f2bb2
clip device fix (#6924)
2024-10-07 00:47:32 +08:00
..
bert.py dtype=float in bert log_softmax for loss and accuracy (#6916) 2024-10-06 11:15:56 -04:00
clip.py clip device fix (#6924) 2024-10-07 00:47:32 +08:00
convnext.py
efficientnet.py remove the magic methods for moving between devices [pr] (#6881) 2024-10-04 20:27:52 +08:00
inception.py Compute FID Score (#6802) 2024-10-01 19:47:58 -04:00
llama.py switch symbolic from old to uops, final PR (#6872) 2024-10-04 16:42:27 +08:00
mask_rcnn.py
resnet.py hotfix: resnet to obj.device 2024-09-06 13:06:02 +08:00
retinanet.py
rnnt.py
t5.py Flux.1 (#6334) 2024-09-24 10:08:04 +08:00
transformer.py
unet.py Sharded SDXL Inference (#6328) 2024-09-21 01:26:43 -04:00
unet3d.py
vit.py