🏫 I’m a senior student at Zhejiang University, majoring in Computer Science and Technology.
🔭 I’m interested in Efficient AI through algorithm–system co-design, focusing on hardware-friendly sparse and quantized module design as well as efficient inference strategies.
🚀 I’m currently a leader of ZJUSCT, a super computing team at Zhejiang University which has won several international super computing competitions.
News
⚡ FlashQLA v0.1.1 released
2026-06-30FlashQLA v0.1.1 has been released, adding intra-card sequence parallelism for the backward pass and SM100 support. Check out the GitHub repository.
🎉 TriAttention accepted to ICML 2026
2026-05-10TriAttention has been accepted to ICML 2026. It is an efficient KV cache compression method for long reasoning in large language models. By leveraging Q/K concentration in pre-RoPE space and a trigonometric series to score key importance, TriAttention matches Full Attention reasoning accuracy on AIME25 while achieving 2.5x higher throughput or 10.7x KV memory reduction. Check out the project page and code!
⚡ FlashQLA v0.1.0 released
2026-04-29FlashQLA v0.1.0 has been released as an open-source high-performance linear attention kernel library for Qwen. Check out the GitHub repository.
Publications
A collection of my research publications and academic papers.
Projects
Selected systems and infrastructure projects around efficient AI, LLM inference, and high-performance computing.
Experience
Research Intern
Research intern in Qwen Infra, working on efficient inference infrastructure for Qwen.
- Contributed to FlashQLA, a high-performance TileLang-based linear attention kernel library for Qwen.
- Focused on efficient GPU kernel design, including fused forward/backward kernels, GDN Chunked Prefill optimization, gate-driven context parallelism, and hardware-friendly reformulation.
Team Leader / Member
Teaching Assistant
- HPC101, Summer 2024 and Summer 2025
- Computer Architecture, Spring 2025
Awards
National Scholarship of China
Recipient
ASC25 Student Supercomputer Challenge
First Prize
ASC25 Student Supercomputer Challenge
Application Innovation Award
5th Linear Solver Algorithm and Performance Optimization Competition
Third Prize
Zhejiang University Scholarship
First Prize
4th Linear Solver Algorithm and Performance Optimization Competition
Second Prize
ISC 2024 Student Cluster Competition (online)
4th Place
ASC24 Student Supercomputer Challenge
Second Prize
Zhejiang University Scholarship
Third Prize
Education
Blog
Visit my personal blog to read about my thoughts, experiences, and technical insights.