Commit Graph

1 Commits (09e491f940844cac17c8a5da7768a1690fc770cd)

Author SHA1 Message Date
baihaochen bc3d137248 [feat] add MLA with decoupled RoPE (DeepSeek-V2 style)
- Add model_minimind_mla.py with Q/KV low-rank compression and decoupled RoPE
- Add kv_norm on kv_latent, q_compress + q_up + q_rope_proj for Q path
- Cache format: (kv_latent, k_rope) tuple, 176 floats/tok/layer (4.4x compression)
- Add --use_mla flag to all training scripts and utils
- Add benchmark_gqa_vs_mla.py script
2026-05-12 18:38:47 +08:00