About

Welcome to My Academic Portfolio

๐Ÿ‘ค About Me

I am a first-year PhD student at University of Illinois Urbana Champaign (UIUC) ๐ŸŽ“ advised by Prof. Minjia Zhang. My research focuses on high performance large-scale LLM training ๐Ÿš€.

๐Ÿ”ฌ Research Interests

My research interests span high-performance computing ๐Ÿ’ป, machine learning systems ๐Ÿค–, and GPU optimization โšก. I focus on:

  • โš™๏ธ Large-scale language model training and inference optimization
  • ๐ŸŽฏ GPU kernel optimization
  • ๐Ÿ—๏ธ HPC

๐ŸŽ“ Education

Ph.D. in Computer Science ๐Ÿ“– (In Progress)
University of Illinois Urbana-Champaign, 2025-Present
Advisor: Prof. Minjia Zhang

BS in Computer Science ๐ŸŽฏ (2021 - 2025)
University of California, San Diego
Advisor: Leon Bergen, Mohan Paturi, Taylor Berg-Kirkpatrick

๐Ÿ“š Publications

Quiet Feature Learning in Algorithmic Tasks ๐Ÿง 
P. Naidu, Z. Wang, L. Bergen, R. Paturi
AAAI 2026 Special Track on AI Alignment (Accepted as Oral Presentation) ๐ŸŽค

The Surprising Soupability of Documents in State Space Models ๐Ÿ“„
Y. Jafari*, Z. Wang*, L. Bergen, T. Berg-Kirkpatrick
arXiv preprint arXiv:2505.24033, 2025

Omniwise: Predicting GPU Kernels Performance with LLMs ๐Ÿ”ฎ
Z. Wang, C. Ramos, M. A. Awad, K. Lowery
In submission to U.S. Patent Application.

Preliminary Results of the MLPerf BERT Inference Benchmark on AMD Instinct GPUs ๐Ÿ“Š
Z. Wang, K. Vu, M. Hodak, A. Mehrotra, F. Gutierrez, K. Smith, G. Seo, et al.
Practice and Experience in Advanced Research Computing 2024: Human Powered Computing, 2024

๐Ÿ† Competitions & Awards

Student Cluster Competition (SC24) - Team Co-Lead, MLPerf Benchmark Lead ๐Ÿฅˆ
SC Conference Series (Mar 2024 - Nov 2024)
Atlanta, Georgia

  • 1st Place Among U.S. & Europe Teams, 4th Place Overall
  • 1st in MLPerf Benchmark in AMD GPUs (2nd in all GPU groups)
  • Designed heterogeneous 3-node system with 2 GPU nodes (1x EPYC Genoa 9534, 4x Instinct MI210) and 1 CPU node (1x EPYC Genoa 9634)
  • Supported first open-source multi-node optimized MIGraphX backend for MLPerf SDXL Inference benchmark

Student Cluster Competition (SC23) - Team Co-Lead, MLPerf Benchmark Lead ๐Ÿฅ‰
SC Conference Series (Mar 2023 - Nov 2023)

  • 3rd Place Overall, 1st Among U.S. Teams
  • 1st in MLPerf Benchmark
  • Designed 3-node enterprise computational cluster with 12 AMD Instinct MI210 GPUs and 6 AMD EPYC 9684x CPUs
  • Represented UCSD competing against worldโ€™s best HPC teams at Supercomputing Conference 23

๐Ÿ’ผ Experience

Internships ๐Ÿ’ก

Performance Engineer (MLPerf Team) โšก
AMD (May 2025 - Aug 2025)
Manager: Miro Hodak
Santa Clara County, California

  • Core contributor to MLPerf Inference 5.1 Llama3 405B Submission
  • Core contributor to MLPerf Training Llama 3 8B training benchmark

Research Intern (RAD Profiling Team) ๐Ÿ”ฌ
AMD (Jun 2024 - Sep 2024)
Manager: Keith Lowery
Austin, Texas

  • Conducted LLM-guided software optimization research under Keith Lowery Group
  • Co-authored two Invention Disclosure Forms (IDFs)
  • In process of US Patent Application

    News & Updates