Peizhen Zheng

Peizhen Zheng

AI Infrastructure Researcher

Huawei Technologies · 2012 Laboratories

About

I am an AI Infrastructure Researcher in the Computational Mathematics Group, Theory Research Department, 2012 Laboratories at Huawei Technologies. My work spans LLM training, inference and data infrastructure.

I independently develop APIE, an LLM inference framework designed for recursive self-improvement (RSI), with optimization skills, benchmarks and validation tools for coding agents. I have also worked on openPangu MTP architecture selection and full-graph inference, DeepSeek performance optimization on Ascend, and distributed pretraining data pipelines. My engineering interests cover GPU/NPU systems, serving, communication overlap and data processing.

I received my master’s degree from Georgia Tech and studied at Zhejiang University’s ZJU-UIUC Institute. My earlier research interests include computer graphics and 3D scene reconstruction.

Download resume · Google Scholar · GitHub

Interests
  • LLM inference & distributed computing
  • RSI & agent-driven optimization
  • Pretraining data infrastructure
  • Computer graphics & 3D reconstruction
Education
  • MS in Computer Science and Engineering, 2024

    Georgia Institute of Technology

  • BS in Civil Engineering, 2022

    Zhejiang University · ZJU-UIUC Institute

Experience

 
 
 
 
 
AI Infrastructure Researcher
February 2025 – Present Shanghai

Computational Mathematics Group, Theory Research Department (Yuwei Fan).

Work on inference engines, MTP, communication optimization and pretraining data infrastructure.

Skills

Platforms — GPU / NPU; PyTorch, torch_npu and Triton; Ascend adaptation and performance optimization.

Inference — Continuous batching, paged KV cache, P/D disaggregation, MTP, fused graphs and asynchronous pipelines; vllm-ascend (ascend-vllm/omni-infer).

Training & parallelism — Familiar with Megatron and PyTorch FSDP; DP / TP / PP / CP / SP / EP; collectives and computation–communication overlap.

Development & data — Python, C++, Rust and Git; Data-Juicer, distributed pipelines, document parsing and code deduplication.

Agent development — Claude Code / Codex / DeepSeek Harness; skills, issue/PR workflows and automated validation.

Languages — TOEFL 107; CET-6.

Contact

Get in touch about LLM infrastructure, efficient inference and agent systems.