Yi Wang
Principal Software Engineer at Apple
Palo Alto, California, United States
Summary
🤩
Rockstar🎓
Top SchoolYi Wang is a Principal Software Engineer with 13 years of experience building large-scale ML and inference systems, currently advancing LLM inference on CUDA, Apple Silicon GPU, and the Apple Neural Engine at Apple. He has led ML infrastructure and distributed training efforts across top tech firms—driving PyTorch distributed work at Meta, founding SQLFlow at Ant Financial to run ML in SQL, and acting as chief scientist for PaddlePaddle at Baidu. His background spans research and production: from distributed LDA and pLDA at Google to Peacock topic modeling at Tencent and productionizing recommendation proofs-of-concept using TorchRec. An active open-source contributor, he has implemented core components like a SQL lexer for SQLFlow and improved tooling in Go+ and Paddle, demonstrating attention to both performance and maintainability. Based in Palo Alto, he pairs a PhD-level ML foundation from Tsinghua with hands-on systems engineering, uniquely bridging algorithmic research and GPU/embedded inference stacks. Notably, he has repeatedly taken research prototypes into production environments for banks, ads, and large recommendation systems.
13 years of coding experience
15 years of employment as a software developer
Ph.D. Machine Learning and Artificial Intelligence, Ph.D. Machine Learning and Artificial Intelligence at Tsinghua UniversityResearch Associate Machine Learning, Research Associate Machine Learning at City University of Hong Kong
go, c++, English, Chinese
Stackoverflow
673reputation
793kreached
12answers
6questions
Github Skills (40)
unit-testing10
thrust10
pytorch10
programming-language10
parser10
c-language10
lexer10
python10
machine-learning10
deeplearning-ai10
go10
deep-learning10
language-design10
tensorflow10
parsing10
Programming languages (18)
Github contributions (5)
sql-machine-learning/sqlflow
Oct 2018 - Oct 2020
Brings SQL and AI together.
Role in this project:
Back-end Developer
Back-end DeveloperContributions:17 reviews, 358 commits, 504 PRs in 2 years
Contributions summary:Yi primarily contributed to the implementation of a SQL lexer in Go for the SQLFlow project. These commits included the addition of a lexer, associated comments, unit tests, and configuration for the .travis.yml file, demonstrating a focus on building a foundational component for parsing and interpreting SQL statements. The user also merged code from the 'develop' branch, which improved and maintained the project.
pythonsql-syntaxdeep-learningsqlmachine-learning
sql-machine-learning/elasticdl
Oct 2018 - Sep 2020
Kubernetes-native Deep Learning Framework
Role in this project:ML Engineer
ML EngineerContributions:9 reviews, 131 commits, 104 PRs in 1 year 11 months
Contributions summary:Yi's commits primarily focused on modifying and extending the ElasticDL framework, which is designed for distributed deep learning on Kubernetes. They worked on integrating a swamp optimization algorithm, and refactoring to use PyTorch for MNIST dataset training. Their contributions involved code modifications to core components, implementing new optimization strategies, and integrating the framework with different ML libraries and datasets. This suggests a strong focus on enhancing the capabilities and flexibility of the ElasticDL for various deep learning tasks.
deep-learningmachine-learningmlopsdistributed-systemskubernetes
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.