Mingkai Zheng

Mingkai Zheng

PhD student

Department of ECE @ Rutgers

Biography

Hi, there! I am Mingkai Zheng, a PhD student from Rutgers University, NJ. I am fortunate to receive supervision from Prof. Zhao Zhang. Most of my research works focus on improving the training efficiency of large language models (LLMs).

Before that, I was a research engineer at Texas Advanced Computing Center (TACC). I received my Master of Engineering degree from the Department of Electrical and Computer Engineering at Cornell University and my Bachelor of Engineering degree from University of Liverpool and Xi’an Jiaotong-Liverpool University.

Interests
  • High Performance Computing (HPC)
  • Large Language Model (LLM)
  • Deep Learning (DL)
Education
  • Ph.D. in Electrical and Computer Engineering, 2024 - 2027 (expected)

    Rutgers University

  • M.Eng. in Electrical and Computer Engineering, 2021 - 2022

    Cornell University

  • B.Eng. in Electrical Engineering, 2016 - 2020

    University of Liverpool

  • B.Eng. in Electrical Engineering, 2016 - 2020

    Xi'an Jiaotong-Liverpool University

Internship

 
 
 
 
 
GenBio AI
Research Scientist Intern
May 2025 – Nov 2025 Palo Alto, CA
 
 
 
 
 
Texas Advanced Computing Center
Research Engineer
Jan 2023 – Jan 2024 Austin, TX

Publications

Quickly discover relevant content by filtering publications.
(2026). Broken Symmetry in BF16 Attention: Why FlashAttention Gradients Blow Up Late in Training. arXiv.

PDF Cite arXiv

(2026). Diamond Agent: Agentic Control of Federated HPC Resources as a Service. arXiv.

PDF Cite arXiv

(2026). A Comparative Study on LLM-enabled HPC User Support. in PEARC 2026.

Cite DOI

(2026). PHOENIX: Resilient LLM Training with Hot-Swapping via Zero-Overhead Checkpoint. arXiv.

PDF Cite arXiv

(2026). SCAPE: Accurate and Efficient LLM Training with Extreme Sparse Communication. arXiv.

PDF Cite arXiv

Projects

*
Movie Ticket Reservation Platform
An online movie ticket reservation platform.
Movie Ticket Reservation Platform
Short/Long-Term Power Load Forecast
Using LSTM with federated learning to predict Cornell’s power load.
Short/Long-Term Power Load Forecast
Image Compression using CUDA
Using CUDA FFT to perform image compressions.
Image Compression using CUDA