Juntai Zheng
Software Engineer at Databricks
California, United States
Summary
🤩
Rockstar🎓
Top SchoolJuntai Zheng is a software engineer with eight years of experience focused on ML platform development, currently contributing to MLflow at Databricks. A UC Berkeley CS graduate based in California, he specializes in integrating TensorFlow models into production tooling—adding save/load support, categorical column handling, and autologging compatibility for TensorFlow 2.0. His work bridges MLOps and developer ergonomics, ensuring models are logged, managed, and reproducible across runs. He began contributing to MLflow during a Databricks internship and has continued to advance the open-source project, demonstrating both deep technical skill and a commitment to community tooling.
7 years of coding experience
Bachelor’s Degree, Computer Science, Bachelor’s Degree, Computer Science at University of California, Berkeley
Chinese, English
Stackoverflow
36reputation
2kreached
1answer
0questions
Github Skills (7)
machine-learning10
mlflow10
tensorflow10
python10
mlops9
model-management9
tkinter6
Programming languages (1)
Github contributions (5)
mlflow/mlflow
Jun 2018 - Mar 2022
Open source platform for the machine learning lifecycle
Role in this project:
MLOps Engineer
MLOps EngineerContributions:8 reviews, 49 commits, 88 PRs in 3 years 10 months
Contributions summary:Juntai primarily contributed to enhancing the MLflow platform, specifically by adding and improving TensorFlow model support. Their work involved implementing TensorFlow model saving and loading functionality, including support for categorical columns and ensuring compatibility with TensorFlow 2.0. They also focused on integrating TensorFlow models with the MLflow autologging feature, ensuring proper run management and logging of parameters, metrics, and model summaries. This demonstrates a focus on integrating ML models with the MLflow platform for ease of use.
pythonlifecyclemlmachine-learningincremental-learning
mlflow/mlflow-apps
May 2018 - Aug 2018
MLflow App Library
Contributions:62 commits, 9 PRs in 2 months
data-sciencemlmlopsmachine-learningapp-library
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.