Tong Shen
Senior Software Engineer at Snowflake
California, United States
Summary
🤩
Rockstar🎓
Top SchoolTong Shen is a Senior Software Engineer with 12 years of experience building high-performance distributed systems and ML infrastructure, currently based in California and working at Snowflake. He has a strong track record at Google driving critical pieces of Search infrastructure, TensorFlow TPU runtime features (including outside compilation and distributed BERT), and internal RPC and clock-synchronization research. Tong contributes to low-level ML compiler work (openxla/xla), improving host computation, sharding, and data transfer semantics—evidence of deep systems and compiler expertise. He combines production-scale reliability (supporting billions of queries per day) with research-grade solutions that enable large-scale ML deployment. A Tsinghua-trained computer scientist, he quietly bridges the gap between cutting-edge research and production engineering, often surfacing optimizations that make ML workloads practically usable at scale.
11 years of coding experience
7 years of employment as a software developer
MS, Computer Science, MS, Computer Science at Tsinghua University
Github Skills (7)
compiler-optimization10
xla10
compiler10
c-language10
cprogramming-language10
data-transfer9
tensorflow8
Programming languages (4)
Github contributions (5)
openxla/xla
Aug 2018 - Jan 2020
A machine learning compiler for GPUs, CPUs, and ML accelerators
Role in this project:
Back-end Developer
Back-end DeveloperContributions:5 commits in 1 year 5 months
Contributions summary:Tong made several contributions related to the XLA compiler's internal operations. These changes include refactoring the handling of host computations by replacing older mechanisms with SendToHost/RecvFromHost pairs and updating metadata management. Additionally, the user removed the HostModule and HostCompute HLO, reflecting a shift in the architecture. Furthermore, the user added attributes and made modifications related to sharding and data transfer.
compilercommunity-drivenmachine-learningmodular
er91/websocket_channel
Sep 2015 - Sep 2015
Contributions:2 pushes, 1 branch in 4 days
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.