Jinliang/fsdp no shard vit#1

Open

wplf wants to merge 4 commits intojinliang/mimo-kimifrom

jinliang/fsdp-no-shard-vit

Commits on Apr 7, 2026

feat: add fsdp_no_shard API to keep params replicated in FSDP

wplf
and
claude
committed

Commits on Apr 8, 2026

perf: skip FSDP hooks for fsdp_no_shard modules

wplf
and
claude
committed
fix: preserve _fsdp_no_shard through meta-device param materialization

wplf
and
claude
committed

Commits on Apr 9, 2026

clean useless code
wplf
committed