Tianrun Li
Software Engineer at Databricks
Sunnyvale, California, United States
Summary
🤩
Rockstar🎓
Top SchoolTianrun Li is a software engineer with 10 years of experience building high-performance ML and data systems, currently working on GenAI platform and performance at Databricks in Sunnyvale. Previously at Google he drove key infrastructure for TensorFlow and JAX—authoring JAX-native training loop components, integrating TPU embedding for large-scale recommendation systems, and implementing DTensor SPMD expanders in the MLIR compiler. He also owned core TPU profiler tooling used by thousands of engineers, with contributions spanning memory analysis, inference latency breakdowns, and chip bring-up validation for TPU v5/v6. Tianrun’s background includes designing OLAP query planners and data modeling layers for billion-query backends, and research work that accelerated database and graph analytics performance. He blends systems-level compiler and profiler expertise with practical ML training and distribution experience, and has a track record of moving research ideas into production at scale.
10 years of coding experience
9 years of employment as a software developer
Graduate Student Computer Science, Graduate Student Computer Science at University of Wisconsin-MadisonBachelor of Engineering (B.E.) Computer Science and Technology, Bachelor of Engineering (B.E.) Computer Science and Technology at Tsinghua University
Chinese, English
Stackoverflow
61reputation
6kreached
0answers
5questions
Github Skills (24)
apache-spark10
container10
python10
lifecycle10
incremental-learning10
machine-learning10
mlflow10
model-management10
dvc10
networking10
ml10
scale10
ai10
serverless10
kubernetes9
Programming languages (2)
Github contributions (5)
lichenran1234/mlflow
Nov 2021 - May 2022
Open source platform for the machine learning lifecycle
Contributions:26 pushes, 5 branches in 6 months
pythonlifecyclemlmachine-learningmachine-learning-lifecycle
lichenran1234/experimental-lda
Oct 2021 - Jul 2024
Contributions:10 pushes in 2 years 10 months
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.