About Me

I am a Ph.D. student in Electronic and Computer Engineering at The Hong Kong University of Science and Technology, advised by Prof. Yuan Xie. I build systems for agentic AI infrastructure: speculative runtimes, memory systems, and serving stacks that make long-horizon LLM agents reliable, efficient, observable, and deployable on real AI clusters.

My work connects systems research with deployable prototypes: agent workflow platforms, trace and evaluation infrastructure, LLM/MoE serving, GPU/CPU-GPU runtime systems, memory-centric AI accelerators, and hardware-software co-design for sparse and irregular AI workloads. Recent work includes TomasuLLM, which brings out-of-order execution to coding agents to reduce time spent waiting on tools. Before joining HKUST, I received my M.S. and B.S. in Microelectronics from Fudan University.

I am open to collaboration on agent runtime systems, AI infrastructure, and efficient model serving across the software-hardware boundary.

Research Interests

  • Agentic AI Infrastructure: speculative execution, runtime, tool use, state management, observability, and evaluation for long-horizon autonomous agents.
  • Agent Memory and Trace Systems: dependency tracing, semantic locality, provenance, target construction, and replayable evaluation for coding agents and research agents.
  • LLM and MoE Serving: CPU-GPU cooperation, KV-cache movement, batching, speculative execution, and transfer-granularity optimization.
  • GPU-Cluster Systems: resource-efficient orchestration for heterogeneous AI clusters and production inference workloads.
  • Hardware-Software Co-Design: memory hierarchy, dataflow, sparse inference, and CIM/PIM acceleration for irregular AI computation.

Recent News [View Publications]

  • 10/2026 Built Megahit Annotation Studio to organize client materials, inspect AI annotations against source evidence, and review suggested relationships.
  • 09/2026 TomasuLLM is on arXiv: out-of-order execution helps coding agents work ahead while tools run.
  • 08/2026 Open-sourced SpecDecode Viz and HKUST Student Expense Agent.
  • 07/2026 Delta-MoE was accepted to ICCAD 2026.
  • 06/2026 SwiftCIM was accepted to ESSERC 2026.
  • 05/2026 STS was posted on arXiv.
  • 04/2026 DS-CIM was accepted to DATE 2026.
  • 11/2025 McPAL appeared in DAC 2025.
  • 08/2025 DIRC-RAG appeared in ISLPED 2025.
  • 07/2024 FullSparse was published at CF 2024.

Selected Projects

A few current projects and research prototypes around agent products, agent traces, LLM serving, and efficient AI systems.

TomasuLLM

Bringing out-of-order execution to coding agents

Long-running compilers, tests, and repository commands leave coding agents waiting. TomasuLLM borrows out-of-order execution and in-order commit from processors to overlap those waits with likely future work. It drafts future actions, runs them in isolated copy-on-write sandboxes, and tracks their dependencies and effects. The runtime validates actions, dependencies, and effects against the committed state before committing accepted results from real executions in the agent's trajectory order.

Evaluation reports 1.31× speedup on 100 SWE-bench Verified tasks, 1.35× on 28 Terminal-Bench 2.0 tasks, and 1.27× matched-progress speedup across 18 SWE-Marathon sessions. No false accepts were observed in 4,010 audited validation records.

Megahit Annotation Studio

Turning expert experience into knowledge you can verify

Expert experience is often scattered across client documents and conversations. Megahit Annotation Studio imports Lark documents and chats alongside local files, organizes materials along each client's business timeline, and uses AI to draft annotations linked to source passages. Reviewers inspect the evidence, correct annotations, and confirm what to keep. Relationships proposed across sources remain suggestions until reviewers check the supporting evidence; human corrections can become approved rules for later annotations.

Building on the earlier GrowthOS consulting workbench, this prototype focuses on reviewing source material and making expert knowledge traceable. The separate workbench supports contextual Chat and human evaluation of answers with and without a knowledge base, keeping the question, conversation context, and model settings fixed.

SpecAgent

Ongoing coding-agent trace workspace

A work-in-progress workspace for organizing coding-agent traces, target sampling, input checks, and small vLLM/Slurm experiments.

AgentMem and AgentProf

Agent memory locality and workload profiling

Research prototypes for studying semantic locality in agent memory and profiling how agent inference differs from ordinary chatbot serving in context growth, KV-cache behavior, idle gaps, and batching.

HETC / LLM Serving Prototypes

Tensor compression and long-context serving experiments

Prototypes for efficient LLM systems, including modular tensor compression for fp16/bf16 tensors and speculation-guided ideas for sparse attention, KV prefetching, and long-context serving.

Open Source Projects

Some recently open-sourced projects that make AI systems easier to inspect and everyday research workflows easier to automate.

SpecDecode Viz

A real-time visualization playground for speculative decoding, showing draft tokens, verification, accepted tokens, and the intuition behind serving speedups. Follow or star the project if this helps you explain or debug speculative decoding.

Usage demo:

HKUST Student Expense Agent

If you are an HKUST student also struggling with tedious reimbursement forms, this agent can automatically upload reimbursement materials and fill the required fields in the HKUST finance system. You can use it as long as you have an LLM API key.

Education

  • Ph.D. in Electronic and Computer Engineering
    The Hong Kong University of Science and Technology, 2024 - Present
    Advisor: Prof. Yuan Xie
  • M.S. in Microelectronics
    Fudan University, 2021 - 2024
  • B.S. in Microelectronics
    Fudan University, 2017 - 2021

Selected Publications [Back to Top]

*: Equal contribution. Local PDF buttons point to full-text files in this repository when an open version is available.

arXiv STS paper first page thumbnail

STS: Efficient Sparse Attention with Speculative Token Sparsity

Jiangnan Yu*, Ceyu Xu*, and Yongji Wu* Co-first author

arXiv preprint, 2026.

PDF arXiv

ICCAD Delta-MoE paper first page thumbnail

Delta-MoE: Rethinking Transfer Granularity for CPU-GPU MoE Serving

Jiangnan Yu, et al. First author

IEEE/ACM International Conference on Computer-Aided Design (ICCAD), 2026. Accepted

PDF

ESSERC SwiftCIM paper first page thumbnail

SwiftCIM: A 55nm 23.2 microJ/Token L-0.5 ReRAM Coupled Digital CIM Accelerator with Fully-Fused Multi-Head Attention Dataflow for FlashAttention

Jiangnan Yu, et al.

ESSERC, 2026. Accepted

PDF

JETCAS Sparse Transformer paper first page thumbnail

Scalable Sparse Transformer Accelerator with In-Memory Butterfly Zero Skipper and Local Attention Reusable Engine for Semi-Structured-Pruned NN

Shiwei Liu, Jiangnan Yu, Peizhe Li, Feng Lin, Chixiao Chen

IEEE Journal on Emerging and Selected Topics in Circuits and Systems (JETCAS), 2026.

PDF DOI

DATE DS-CIM paper first page thumbnail

DS-CIM: Digital Stochastic Computing-In-Memory Featuring Accurate OR-Accumulation via Sample Region Remapping for Edge AI Models

Kunming Shao*, Liang Zhao*, Jiangnan Yu*, Zhipeng Liao, Xiaomeng Wang, Yi Zou, Tim Kwang-Ting Cheng, Chi-Ying Tsui Co-first author

Design, Automation and Test in Europe Conference (DATE), 2026.

PDF arXiv DOI

DAC McPAL paper first page thumbnail

McPAL: Scaling Unstructured Sparse Inference with Multi-Chiplet HBM-PIM Architecture for LLMs

Shiwei Liu, Zhirui Huang, Jiangnan Yu, Qi Liu, Chixiao Chen

Design Automation Conference (DAC), 2025.

PDF DOI

ISLPED DIRC-RAG paper first page thumbnail

DIRC-RAG: Accelerating Edge RAG with Robust High-Density and High-Loading-Bandwidth Digital In-ReRAM Computation

Kunming Shao*, Zhipeng Liao*, Jiangnan Yu*, Liang Zhao, Qiwei Li, Xijie Huang, Jingyu He, Fengshi Tian, Yi Zou, Xiaomeng Wang, Tim Kwang-Ting Cheng, Chi-Ying Tsui Co-first author

IEEE/ACM International Symposium on Low Power Electronics and Design (ISLPED), 2025.

PDF arXiv IEEE

CF FullSparse paper first page thumbnail

FullSparse: A Sparse-Aware GEMM Accelerator with Online Sparsity Prediction

Jiangnan Yu, Yang Fan, Hanfei Wang, Yecheng Zhang, Chixiao Chen, Xiankui Xiong, Dong Xu, Fan Yang First author

ACM International Conference on Computing Frontiers (CF), 2024.

PDF DOI

GLSVLSI TPNoC paper first page thumbnail

TPNoC: An Efficient Topology Reconfigurable NoC Generator

Jiangnan Yu, Yang Fan, Xiaoling Yi, Chixiao Chen, Jun Tao, Dong Xu, Xiankui Xiong, Haitao Yang First author

ACM Great Lakes Symposium on VLSI (GLSVLSI), 2023.

PDF DOI

ISCAS NNASIM paper first page thumbnail

NNASIM: An Efficient Event-Driven Simulator for DNN Accelerators with Accurate Timing and Area Models

Xiaoling Yi, Jiangnan Yu, Zhenyu Wu, Xiankui Xiong, Dong Xu, Chixiao Chen, Jun Tao, Fan Yang

IEEE International Symposium on Circuits and Systems (ISCAS), 2022.

PDF DOI

Misc

Motto: With effort, what you wish for can come true.

I enjoy running, hiking, and building small tools that make research work less fragile.