Zhihao Bai
Research Scientist at ByteDance
Bellevue, Washington, United States
Summary
👤
Senior🎓
Top SchoolZhihao Bai is a Research Scientist at ByteDance’s Machine Learning Systems group with nine years of experience bridging systems research and practical ML infrastructure. He earned a PhD in Computer Science from Johns Hopkins University (2022) and has published at top venues including OSDI, S&P and FAST, where his DistCache work won a Best Paper Award. Zhihao’s internships at ByteDance, Alibaba, and Microsoft Research Asia reflect a strong track record of turning research prototypes—like PipeSwitch and AI-driven network planning—into deployable systems. Based in Bellevue, WA, he focuses on performance-oriented machine learning systems and cross-layer network solutions, often combining systems-level insights with applied ML techniques. An understated strength is his ability to navigate both academic rigor and product-driven engineering, accelerating DL workloads in production environments.
9 years of coding experience
Johns Hopkins University上海交通大学
Github Skills (15)
context-switching10
deep-learning8
machine-learning8
pytorch8
twitch8
chat7
spatial7
in-memory5
big-data-analytics5
spark5
apache-spark5
data-analytics5
analytics4
big-data3
computer-networks3
Programming languages (2)
Github contributions (5)
netx-repo/PipeSwitch
Nov 2020 - Oct 2021
PipeSwitch: Fast Pipelined Context Switching for Deep Learning Applications
Contributions:201 commits in 11 months
pytorchdeep-learningmachine-learningswitchingcontext-switching
baizh1994/PipeSwitch
Jul 2021 - Oct 2021
PipeSwitch
Contributions:454 commits, 453 pushes, 78 branches in 3 months
twitchchat
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.