chenyuandGitHub 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

unet3d.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00

unet.py

Stable Diffusion model init for mlperf (#12314)

2025-10-02 02:28:41 -04:00

vit.py

move to new cached fetch (#2493)

2023-11-28 17:36:55 -08:00