MCPcopy Index your code
hub / github.com/MuLabPKU/TransArch

github.com/MuLabPKU/TransArch @main

Chat with this repo
repository ↗ · DeepWiki ↗ · + Follow
379 symbols 1,171 edges 39 files 87 documented · 23%
What it actually does AI analysis from the code graph — generated when you open this
loading…
README

TransArch: Hardware-Aware Architecture Transfer for Foundation Models

Modern large language models are increasingly bottlenecked by communication rather than computation on today's hardware. TransArch is a collection of research projects that design hardware-friendly model architectures and migrate existing pre-trained models into these new architectures with minimal performance loss.

📦 Projects

Project Venue Description Paper
CLOVER ICML 2025 Cross-layer SVD pruning of Q-K and V-O pairs in attention heads; combined with TransMLA achieves up to 11.1× speedup over LLaMA-2-7B arXiv
TransMLA NeurIPS 2025 Spotlight Convert GQA models (LLaMA, Qwen, Mixtral, …) to DeepSeek-MLA with full Absorb compatibility and up to 10.6× inference speedup arXiv
TPLA ASPLOS 2026 Tensor Parallel Latent Attention — partitions latent representations across devices, achieving 1.79×/1.93× speedup on DeepSeek-V3/Kimi-K2 arXiv
HISA Preprint Hierarchical two-stage indexer for fine-grained sparse attention, achieving 2×–4× speedup at 32K–128K context arXiv
MISA Preprint Mixture-of-experts routing over DSA indexer heads — matches dense DSA with 8×/4× fewer indexer heads and ~3.82× kernel speedup on H200 arXiv
GQLA Preprint Group-Query Latent Attention — one set of weights, two decoding paths (MQA-absorb for H100, GQA+MTP for H20), with up to 8-way TP arXiv

📰 News

  • [2026.05] GQLA code released: see GQLA_preprint/.
  • [2026.05] GQLA preprint released: arXiv:2605.15250.
  • [2026.05] MISA preprint released: arXiv:2605.07363.
  • [2026.03] HISA preprint released: arXiv:2603.28458.
  • [2026.02] 🎉 TransMLA is adopted by Ant Group's latest 1T model Ling-2.5-1T! This demonstrates the robust scalability of TransMLA in ultra-large-scale LLMs.
  • [2025.11] TPLA accepted at ASPLOS 2026 (Summer cycle).
  • [2025.09] TransMLA accepted at NeurIPS 2025 (Spotlight, Top 3.19%).
  • [2025.05] CLOVER accepted at ICML 2025.

📋 To-Do

  • [ ] Release TPLA code
  • [ ] Release HISA code
  • [ ] Release MISA code

📚 Citation

If you find our work useful, please cite the relevant paper(s):

@inproceedings{meng2025transmla,
  title={TransMLA: Multi-head Latent Attention Is All You Need},
  author={Meng, Fanxu and Tang, Pingzhi and Tang, Xiaojuan and Yao, Zengwei and Sun, Xing and Zhang, Muhan},
  booktitle={Advances in Neural Information Processing Systems (NeurIPS)},
  year={2025}
}

@inproceedings{meng2025clover,
  title={CLOVER: Cross-Layer Orthogonal Vectors Pruning and Fine-Tuning},
  author={Meng, Fanxu and Tang, Pingzhi and Jiang, Fan and Zhang, Muhan},
  booktitle={International Conference on Machine Learning (ICML)},
  year={2025}
}

@inproceedings{tang2026tpla,
  title={TPLA: Tensor Parallel Latent Attention for Efficient Disaggregated Prefill and Decode Inference},
  author={Tang, Xiaojuan and Meng, Fanxu and Tang, Pingzhi and Wang, Yuxuan and Yin, Di and Sun, Xing and Zhang, Muhan},
  booktitle={International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS)},
  year={2026}
}

@article{xu2026hisa,
  title={HISA: Efficient Hierarchical Indexing for Fine-Grained Sparse Attention},
  author={Xu, Yufei and Meng, Fanxu and Jiang, Fan and Wang, Yuxuan and Zhou, Ruijie and Wang, Zhaohui and Wu, Jiexi and Pan, Zhixin and Tang, Xiaojuan and Pei, Wenjie and Liu, Tongxuan and Yin, Di and Sun, Xing and Zhang, Muhan},
  journal={arXiv preprint arXiv:2603.28458},
  year={2026}
}

@article{zhou2026misa,
  title={MISA: Mixture of Indexer Sparse Attention for Long-Context LLM Inference},
  author={Zhou, Ruijie and Meng, Fanxu and Xu, Yufei and Liu, Tongxuan and Lu, Guangming and Zhang, Muhan and Pei, Wenjie},
  journal={arXiv preprint arXiv:2605.07363},
  year={2026}
}

@article{meng2026gqla,
  title={GQLA: Group-Query Latent Attention for Hardware-Adaptive Large Language Model Decoding},
  author={Meng, Fanxu},
  journal={arXiv preprint arXiv:2605.15250},
  year={2026}
}

⭐ Star History

Star History Chart

Core symbols most depended-on inside this repo

yarn_get_mscale
called by 5
TransMLA_NeurIPS_2025/models/deepseek_v2_lite/modeling_deepseek.py
rotate_half
called by 4
TransMLA_NeurIPS_2025/transmla/transformers/deepseek_v3/modeling_deepseek_v3.py
evaluate_ppl
called by 4
CLOVER_ICML_2025/src/pca_calc.py
fit_per_group_pca
called by 4
GQLA_preprint/src/compression.py
evaluate_ppl
called by 3
TransMLA_NeurIPS_2025/transmla/utils.py
get_qkv_calibrate_outputs
called by 3
TransMLA_NeurIPS_2025/transmla/utils.py
partial_rope_freqfold
called by 3
TransMLA_NeurIPS_2025/transmla/partial_rope.py
apply_rotary_pos_emb_interleave
called by 3
TransMLA_NeurIPS_2025/transmla/transformers/deepseek_v3/modeling_deepseek_v3.py

Shape

Method 166
Function 136
Class 76
Route 1

Languages

Python100%

Modules by API surface

TransMLA_NeurIPS_2025/models/deepseek_v2_lite/modeling_deepseek.py75 symbols
TransMLA_NeurIPS_2025/transmla/transformers/deepseek_v3/modeling_deepseek_v3.py49 symbols
TransMLA_NeurIPS_2025/transmla/vllm_registry/deepseek.py27 symbols
GQLA_preprint/src/vllm_model.py25 symbols
GQLA_preprint/src/compression.py24 symbols
TransMLA_NeurIPS_2025/transmla/utils.py20 symbols
TransMLA_NeurIPS_2025/throughput_test/benchmark_dataset.py14 symbols
CLOVER_ICML_2025/src/pca_calc.py12 symbols
TransMLA_NeurIPS_2025/transmla/partial_rope.py11 symbols
TransMLA_NeurIPS_2025/lighteval/tasks.py9 symbols
TransMLA_NeurIPS_2025/throughput_test/benchmark_throughput.py8 symbols
GQLA_preprint/src/modeling.py8 symbols

For agents

$ claude mcp add TransArch \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact