Whole-workspace rustfmt pass picked up while iterating on Mamba GPU
backward work. Verified formatting-only via diff sampling; no logic
changed.
Co-Authored-By: Claude Sonnet 5 <[email protected]>
- MedusaHeads: K FFN draft heads (SiLU 2-layer); tree candidate generation
via cartesian product of per-head top-k; path verification with oracle;
CE training loss per head (arXiv:2401.10774); 23 tests
- ModelMerger: TIES (task-vector trim+elect-sign+disjoint-merge,
arXiv:2306.01708) + DARE sparse rescaling (arXiv:2311.03099); linear
merge baseline; 29 tests
- MoDLayer/MoDStack: per-token capacity routing (top-k by router score);
residual bypass for skipped tokens; load-balancing aux loss; flops_reduction
= product of capacity_fractions (arXiv:2404.02258); 22 tests
Co-Authored-By: Claude Sonnet 4.6 <[email protected]>