Education

• 

  B.S.E. in Computer Engineering

  College of Engineering

  University of Michigan

  GPA: 4.0/4.0

  2024 - 2026
• 

  B.S. in Mechanical Engineering

  University of Michigan-Shanghai Jiao Tong University Joint Institute

  Shanghai Jiao Tong University, Shanghai

  Core GPA: 3.65/4.0

  2022 - 2026

Honors and Awards

• • Summer Undergraduate Research in Engineering (SURE) Stipend, University of Michigan
  05/2025
• • Dean's Honor List at College of Engineering, University of Michigan
  12/2024, 05/2025

Research Experience

• TurnRL: RL-Driven Turn-Restricted Adaptive Routing for Chiplet-Based Systems
  
  • Advisor: Nathaniel Bleier
  • Developed fast turn-addition algorithms and baselines for arbitrary topologies.
  • Applied Deep Q-Learning with GNN to learn optimal turn sets under different traffic patterns.
  • Evaluated with gem5 Garnet and PARSEC workloads; demonstrated robust latency reduction and scalability.
  09/2024 - 11/2025
• Double-P: Hierarchical Top-P Sparse Attention for Long-Context LLMs
  
  • Advisors: Jishen Zhao, Zhijian Liu
  • Proposed a hierarchical sparse attention framework that jointly optimizes top-p accuracy, selection overhead, and sparse attention cost.
  • Performed coarse-grained top-p estimation at the cluster level using size-weighted centroids, then adaptively refined computation through a second top-p stage.
  • Achieved near-zero accuracy drop with up to 1.8× attention computation reduction and 1.3× end-to-end decoding speedup over state-of-the-art methods.
  09/2025 - 1/2026
• Long-Context LLM Inference Acceleration with ANNS Search
  
  • Advisors: Jishen Zhao, Zhijian Liu
  • Proposed a CPU-GPU heterogeneous ANN pipeline for efficient KV-cache retrieval.
  • Exploited inter-layer query similarity via cross-layer cluster reuse, forming a cross-layer search pipeline.
  05/2025 - 10/2025
• 3D Multi-Physics Modeling of Li-ion Batteries
  
  • Advisor: Yanming Wang
  • Built a comprehensive 3D electrochemical-thermal-mechanical model of lithium-ion cells.
  • Predicted battery aging behavior, enabling more accurate performance and longevity assessment.
  02/2023 - 02/2024

Publications

• TurnRL: RL-Driven Turn-Restricted Adaptive Routing for Chiplet-Based Systems
  
  • K. Zhang*, J. Wang*, R. Dreslinski, N. Bleier.
  • TurnRL: Reinforcement-Learning Driven Turn-Restricted Adaptive Routing for Chiplet-Based System.
  • DAC 2026, under review.
  2026
• Double-P: Hierarchical Top-P Sparse Attention for Long-Context LLMs
  
  • W. Ni, K. Zhang, Z. Yu, O. Nelson, M. Lee, H. Cai, F. Porikli, J. Kim, Z. Liu, J. Zhao.
  • Double-P: Hierarchical Top-P Sparse Attention for Long-Context LLMs.
  • ICML 2026, Under review.
  2026
• Mozart: Ecosystem-Accelerator Codesign Framework
  
  • H. Jin, J. Yang, Y. Liu, B. Lyu, K. Zhang, N. Bleier.
  • Mozart: An Ecosystem-Accelerator Codesign Framework for Composable Heterogeneous Chiplet Based Neural Network Accelerators.
  • ASPLOS 2026, under review.
  2026

Intellectual Property

• PyBatSim V1.0
  Multi-physics Simulation Software for Li-ion Cells
  • Developed a 3D coupled electrochemical-thermal-mechanical simulation tool for lithium-ion cells.
  • Registered as a software copyright with the China National Copyright Administration.
  10/2024

Course Projects

• EECS 470
  Out-of-Order RISC-V Superscalar Processor
  • Advisors: Krisztian Flautner, Ronald G. Dreslinski
  • Proposed framework for RTL design on MIPS R10K microarchitecture with RV32-IM ISA.
  • Developed advanced features: N-way superscalar execution, early branch resolution, early tag broadcast, prefetching, non-blocking associative caches, and a cycle-accurate simulator.
  • Integrated advanced branch prediction subsystems: TAGE, perceptron, and CNN-based predictors, along with a BTB and RAS.
  • Achieved an average CPI of 1.5 under 10ns clock; final system ranked top 2 of the course.
  02/2025 - 05/2025
• EECS 570
  Directory-Based MSI/MESI Cache Coherence Protocol
  • Advisor: Satish Narayanasamy
  • Designed latency-optimized three-hop MSI and MESI protocols with transient states for pending data transfers and acknowledgments.
  • Introduced Exclusive (E) state enabling silent upgrades to Modified without network traffic.
  • Implemented out-of-order message receive support for robust handling of race conditions.
  • Verified correctness using Murphi model checking with coherence, mutual exclusion, and value consistency invariants.
  11/2025
• EECS 570
  Parallel Global Router for VLSI
  • Advisor: Satish Narayanasamy
  • Developing a parallelized global router for VLSI designs using CPU and GPU acceleration.
  • Implementing Steiner tree construction, maze routing, and 3-bend routing algorithms with performance optimization.
  • Evaluating on ISPD benchmark circuits for runtime scalability and routing quality metrics.
  Ongoing
• EECS 573
  Adaptive Feature Selection for Heterogeneous Core Scheduling
  • Advisor: Todd Austin
  • Extended GEM5/QEMU to extract comprehensive MICA features and modeled realistic migration costs.
  • Designed an information-theoretic feature selection framework to identify Pareto-optimal MICA subsets balancing accuracy and overhead.
  Ongoing
• EECS 471
  GPU Optimization of CNN Forward Pass
  • Advisor: Reetuparna Das
  • Implemented the forward pass of a convolutional neural network (CNN) layer in CUDA with performance-critical kernels.
  • Applied shared memory tiling, register-level GEMM transformations, and loop unrolling, reducing runtime from 13.5s to 0.11s.
  03/2025 - 04/2025

Coursework (blue = graduate standing)