About Me

Hi! I’m Zhiwen Mo (莫志文), a first-year PhD student in the Department of Computing at Imperial College London, co-advised by Dr. Hongxiang Fan and Prof. Wayne Luk. My PhD is funded by the Scaling Compute project via ARIA.

Previously, I was a research intern at Microsoft Research Asia (MSRA), supervised by Dr. Lingxiao Ma and Dr. Jilong Xue, with generous help from Dr. Shijie Cao and Dr. Fan Yang.

I earned my Bachelor’s degree from the School of Microelectronics at Shanghai Jiao Tong University, ranking 1st with a GPA of 3.96/4.3.

I’m broadly interested in ML hardware and systems, particularly in GPU microarchitecture, performance modeling, kernel optimization, and hardware-software co-design.

I’m a member of Tile-AI TileAI.

Email: zhiwen.mo25@ic.ac.uk


📰 News


🎓 Education

  • PhD in Computing, Imperial College London
    2025–2028 (expected) · Funded by Scaling Compute (ARIA)
  • PhD in Electronic Engineering, Shanghai Jiao Tong University (withdrawn)
    2022–2024 Zhiyuan Honor PhD Program, ~top 2%
  • B.Eng. in Microelectronics, Shanghai Jiao Tong University
    2018–2022 · GPA: 3.96/4.3 (Ranked 1st)

💼 Internship

2026–Present

Qwen Team – Research intern, AI Infrastructure

  • Distributed LLM inference optimization
Qwen Logo
2025

Tile-AI – Research intern, 3D-Stacked Architecture

  • Architectural design space exploration for 3D-Stacked accelerators
Tile AI Logo
2023–2025

Microsoft Research Asia – System Research Group

Microsoft Logo
2022–2023

Shanghai AI Laboratory – SoC Design

  • Early-stage investigation for wafer-scale chips
Shanghai AI Lab Logo

📄 Selected Publications

ISCA
LUT Tensor Core: A Software-Hardware Co-Design for LUT-Based Low-Bit LLM Inference

Zhiwen Mo, Lei Wang, Jianyu Wei, Zhichen Zeng, Shijie Cao, Lingxiao Ma, Naifeng Jing, Ting Cao, Jilong Xue, Fan Yang, Mao Yang

The 52nd International Symposium on Computer Architecture (ISCA'25), 2025
2025
Internal
A GPU Kernel Research Project

Zhiwen Mo, et al.

Research in progress, 2025
2025
ASPLOS
Democratizing Agentic AI with Fast Test-Time Scaling on the Edge

Hao (Mark) Chen, Zhiwen Mo, Guanxi Lu, Shuang Liang, Lingxiao Ma, Wayne Luk, Hongxiang Fan

The31st ACM International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS'26), 2026
2026
OSDI
PipeThreader: Software-Defined Pipelining for Efficient DNN Execution

Yu Cheng, Lei Wang, Yining Shi, and Yuqing Xia, Lingxiao Ma, Jilong Xue, Yang Wang, Zhiwen Mo, Feiyang Chen, Fan Yang, Mao Yang, Zhi Yang

The 19th USENIX Symposium on Operating Systems Design and Implementation (OSDI'25), 2025
2025
arXiv
TileLang: A Composable Tiled Programming Model for AI Systems

Lei Wang, Yu Cheng, Yining Shi, Zhengju Tang, Zhiwen Mo, Wenhao Xie, Lingxiao Ma, Yuqing Xia, Jilong Xue, Fan Yang, Zhi Yang

arXiv preprint arXiv:2504.17577, 2025
2025

👉 See full list at Publications

🏆 Honors & Awards

  • ARIA PhD Scholarship, Imperial College London (£22,780/year, 2025–2028)
  • Chinese National Scholarship, Ministry of Education (2020–2022)
  • SenseTime Scholarship, 30 people/year nationwide in AI (2021)
  • Excellent Graduate of Shanghai, awarded by Shanghai (2022)
  • First Prize, Arm Cup (East China Division) (2021)

🛠 Technical Skills

  • Hardware: Verilog, SystemVerilog
  • Software: Python, C/C++, CUDA (incl. profiling with ncu/nsys), ROCm (incl. rocprof/roccompute), MATLAB

🎨 Misc

Aside from Computer Science:

  • 🎹 Classical music lover (with perfect pitch 🎶⭐)
  • ⚽ Footballer (center back)
  • 🚴 Cycling enthusiast (hands-free rider 😄)
  • ☕ Coffee lover
  • 🎮 Hardcore fan of HoYoverse/miHoYo

Thanks for stopping by!

Cyrene
Cyrene from HSR.