October 28–31, 2019

How to track and manage TensorFlow 2.0 and Keras model experiments with MLflow

Juntai Zheng (Databricks)
5:00pm5:40pm Wednesday, October 30, 2019
Location: Grand Ballroom C/D



ML development brings many new complexities beyond the traditional software development lifecycle. Unlike in traditional software development, ML developers want to try multiple algorithms, tools, and parameters to get the best results, and they need to track this information to reproduce their work. In addition, developers need to use many distinct systems to productionize models. To address these problems, many companies are building custom “ML platforms” that automate this lifecycle, but even these platforms are limited to a few supported algorithms and to each company’s internal infrastructure.

Juntai Zheng introduces MLflow, a new open source project from Databricks that aims to design an open ML platform where organizations can use any ML library and development tool of their choice to reliably build and share ML applications. MLflow introduces simple abstractions to package reproducible projects, track results, and encapsulate models that can be used with many existing tools, accelerating the ML lifecycle for organizations of any size.

With a short demo and code example, you’ll see a complete ML model lifecycle for TensorFlow 2.0 and Keras.

Prerequisite knowledge

  • A basic understanding of Python and machine learning

What you'll learn

  • See a demonstration of the complete ML model lifecycle with TensorFlow 2.0 and Keras
  • Understand MLflow concepts and abstractions for models, experiments, and projects; using tracking APIs during model training; using MLflow UI to visually compare experimental runs with different tuning parameters and evaluate metrics; and integration with TensorBoard
Photo of Juntai Zheng

Juntai Zheng


Juntai Zheng is a software engineer at Databricks and is a member of the team developing MLflow. He has actively contributed to MLflow since its inception, including the TensorFlow support for MLflow projects. He develops the support for TensorFlow 2.0 in MLflow. Juntai holds a bachelor of arts degree from UC Berkeley in computer science.

Leave a Comment or Question

Help us make this conference the best it can be for you. Have questions you'd like this speaker to address? Suggestions for issues that deserve extra attention? Feedback that you'd like to share with the speaker and other attendees?

Join the conversation here (requires login)


Adrian Moldovan |
10/03/2019 9:25pm PDT

Ideally I would like to find out more about low latency deployments and setups, one that autoscale and have a fast response time for generating forecasts.
Also, how this can be integrated in an AutoML setup along with other frameworks, including custom ones.

Contact us


For conference registration information and customer service


For more information on community discounts and trade opportunities with O’Reilly conferences


For information on exhibiting or sponsoring a conference


For media/analyst press inquires