tinygrad/extra/models
chenyu dcee90aa3f
remove requires_grad use in extra/examples (#16238)
except the ones fed into optimizer
2026-05-16 18:40:26 -04:00
..
bert.py remove requires_grad use in extra/examples (#16238) 2026-05-16 18:40:26 -04:00
clip.py move more tests to test/null, split some existing ones (#14512) 2026-02-03 20:20:20 +08:00
convnext.py remove Tensor.no_grad, it's meaningless now [pr] (#10556) 2025-05-28 22:20:02 -07:00
efficientnet.py remove the magic methods for moving between devices [pr] (#6881) 2024-10-04 20:27:52 +08:00
inception.py don't hardcode weights path (#12171) 2025-09-15 00:33:47 -04:00
llama.py llama3: correctly shard wqkv (#14978) 2026-02-23 23:57:10 -08:00
mask_rcnn.py remove requires_grad use in extra/examples (#16238) 2026-05-16 18:40:26 -04:00
resnet.py model/test fix that failed with WEBGPU=1 DEBUG=2 (#14706) 2026-02-12 09:08:16 -05:00
retinanet.py RetinaNet INITMLPERF support (#9950) 2025-04-21 10:32:05 -04:00
rnnt.py remove requires_grad use in extra/examples (#16238) 2026-05-16 18:40:26 -04:00
t5.py Flux.1 (#6334) 2024-09-24 10:08:04 +08:00
transformer.py _one_hot_along_dim input needs to be int (#9179) 2025-02-20 09:00:43 -05:00
unet.py Stable Diffusion model init for mlperf (#12314) 2025-10-02 02:28:41 -04: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