Tianle Zhong | University of Virginia

About Me

Greetings! I am a fourth-year Ph.D. student in Computer Science advised by Prof. Geoffrey Fox. I also work closely with Prof. Keren Zhou.

Before joining UVA, I received my B.Eng. in Computer Science and B.Sc. in Applied Mathematics from the University of Electronic Science and Technology of China in 2022.

I am currently a Research Scientist Intern at ByteDance Seed LLM Infra, where I work on open-source systems including VeOmni, VeXact, and VeRL, as well as internal infrastructure for scalable multimodal MoE pre-training and RL. I also contribute to FlashAttention and Triton. During my undergraduate studies, I interned at the Systems & Networking Research Group, Microsoft Research Asia.

I also spent one year in Japan as an exchange student at The University of Electro-Communications, Tokyo, where I was supervised by Prof. Yoichi Haneda at UEC Sound Media Lab.

Research Interests

Systems for ML: data management for ML, distributed ML systems, HPC for ML
System-ML Co-design: new ML paradigms efficient with hardware

News

[May. 2026] VeXact is open source.
[Nov. 2025] Proton is accepted at CGO’26.
[Oct. 2025] Won the 1st place in GPU MODE Hackathon with $10,000 Nebius GPU credits!
[May 2025] I have been selected for MLSys Student Travel Grant. Thank you MLSys committee!
[May 2025] I will present Youmu at MLSys 2025 in Santa Clara!
[Apr. 2024] I will present Reimu at AIOps workshop of ASPLOS 2024 in San Diego!

Selected Publications

arXiv

Diagnosing Training Inference Mismatch in LLM Reinforcement Learning

Tianle Zhong, Neiwen Ling, Yifan Pi, Zijun Wei, Tianshu Yu, Geoffrey Fox, Peng Wu, Xiao Yu

arXiv preprint arXiv:2605.14220, 2026

PDF Code BibTex

CGO

Proton: Towards Multi-level, Adaptive Profiling for Triton

Keren Zhou, Tianle Zhong, Hao Wu, Jihyeong Lee, Yue Guan, Yufei Ding, Corbin Robeck, Jeff Niu, Phil Tillet

Code Generation and Optimization (CGO), 2026

PDF Code BibTex

MLSys

Youmu: Efficient Columnar Data Pipeline for LLM Training

Tianle Zhong, Jiechen Zhao, Qiang Su, Geoffrey Fox

Eighth Conference on Machine Learning and Systems (MLSys), 2025

PDF Slides BibTex Poster

AIOps

Reimu: Optimizing Data I/O for LLM Datasets on Remote Storage

Tianle Zhong, Jiechen Zhao, Xindi Guo, Qiang Su, Geoffrey Fox

Cloud Intelligence/AIOps Workshop (Co-located with ASPLOS'24).

PDF BibTex

ICASSP

Spherical Convolutional Recurrent Neural Network for Real-Time Sound Source Tracking

Tianle Zhong, Israel Mendoza Velzquez, Yi Ren, Hctor M. Prez Meana, Yoichi Haneda

IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), 2022.

PDF BibTex