Update tests for MoE and parallel optimizations [skip-build]

Apply suggestions from code review

Support multi-component expert_keys (e.g. "experts.w1")

Use component-level matching for expert_keys to avoid shared_experts collision

Normalize parameter FQNs to handle torch.compile / checkpoint wrappers