Zhuoran Zhao
Staff Research Scientist, PyTorch at Meta
Menlo Park, California, United States
Summary
🤩
Rockstar🎓
Top SchoolZhuoran Zhao is a Staff Research Scientist at Meta in Menlo Park who leads PyTorch GPU inference runtime and compiler efforts, with eight years of experience building production ML serving infrastructure. He holds a PhD in Computer Engineering from UT Austin and combines systems research rigor with hands-on C/C++ and Python engineering. At Meta he drove GPU enablement and massive adoption for Ads ML serving and now focuses on PyTorch performance and cross-hardware compatibility, contributing optimizations to core tensor ops and enabling AOTInductor compilation for AMD/HIP and MI300. His open-source contributions to the widely used pytorch/pytorch repo include fast-path implementations like torch.cat and fixes for hardware-specific compilation issues. Quirkily, he recently learned PHP, reflecting a broad curiosity that balances academic depth with pragmatic production fixes.
8 years of coding experience
11 years of employment as a software developer
Master's degree, Computer Engineering, Master's degree, Computer Engineering at The University of Texas at AustinBachelor's degree, Electrical Engineering, Bachelor's degree, Electrical Engineering at Zhejiang University
Github Skills (12)
hip10
pytorch10
machine-learning10
tensor10
gpu10
amd10
autograd9
deep-learning9
c-language8
cuda8
c-programming-language8
python7
Programming languages (3)
Github contributions (5)
pytorch/pytorch
Aug 2023 - Mar 2025
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Role in this project:
ML Engineer
ML EngineerContributions:11 reviews, 20 PRs, 14 pushes in 1 year 7 months
Contributions summary:Zhuoran primarily contributes to the PyTorch codebase, focusing on optimizations and bug fixes related to tensor operations and the AOTInductor compilation backend. Their work includes improving the `torch.cat` implementation, adding fast paths, and addressing issues specific to AMD hardware, including integrating with HIP. They also work on enabling AOTInductor compilation for AMD and fixing related compilation issues, especially for MI300. These changes are primarily focused on performance and compatibility for different hardware backends.
pythongpu-accelerationdeep-learninggpunumpy
SLAM-Lab/NoSSim
May 2017 - Jun 2019
Network-of-Systems Simulator
Contributions:15 commits, 2 PRs, 11 pushes in 2 years 1 month
simulatorsimulationgazebosim
Find and Hire Top DevelopersWe’ve analyzed the programming source code of over 60 million software developers on GitHub and scored them by 50,000 skills. Sign-up on Prog,AI to search for software developers.