feat: 🎸 add vit_tp_exp: DeiT-S tensor-parallel example, cross-checked… - #30

Open

zengarden wants to merge 4 commits into

mainfrom

feat/vit-tp-exp

Commits on Sep 11, 2026

feat: 🎸 add vit_tp_exp: DeiT-S tensor-parallel example, cross-checked L1-L4 against official DeiT

Zeming LI
and
claude
committed
7dfd57d

Commits on Sep 13, 2026

fix: 🐛 preserve input tensors during distributed reductions
Zeming LI
committed
030422d
chore: 🤖 update ViT TP usage and remove platform scripts
Zeming LI
committed
ff68bd9
refactor: 💡 reuse timm ViT components for tensor parallelism
Zeming LI
committed
4e334e1