Presented By O’Reilly and Intel AI
Put AI to work
Sep 4-5, 2018: Training
Sep 5-7, 2018: Tutorials & Conference
San Francisco, CA

Distributed TensorFlow training using Keras and Kubernetes

Magnus Hyttsten (Google), Priya Gupta (Google)
11:55am-12:35pm Thursday, September 6, 2018
Implementing AI
Location: Imperial A
Secondary topics:  Deep Learning tools, Platforms and infrastructure
Average rating: ***..
(3.00, 1 rating)

Who is this presentation for?

  • Developers and researchers

Prerequisite knowledge

  • A basic understanding of machine learning models and TensorFlow or another ML framework

What you'll learn

  • Understand why GPUs and TPUs are effective for ML workloads
  • Learn how to set up a state-of-the-art TensorFlow distributed training cluster using the latest features and following best practices


Magnus Hyttsten and Priya Gupta demonstrate how to perform distributed TensorFlow training using the Keras high-level APIs. They walk you through TensorFlow’s distributed architecture, how to set up a distributed cluster using Kubeflow and Kubernetes, and how to distribute models created in Keras. Along the way, you’ll discover why TPUs and GPUs are so effective at processing machine learning workflows and learn how to configure TensorFlow to use them.

Photo of Magnus Hyttsten

Magnus Hyttsten


Magnus Hyttsten is a developer advocate for TensorFlow at Google, where he works on developing the TensorFlow product. A developer fanatic, Magnus is an appreciated speaker at major industry events such as Google I/O, the AI Summit, AI Conference, ODSC, GTC, QCon, and others on machine learning and mobile development. Right now, he’s focusing on reinforcement learning models and making model inference effective on mobile.

Photo of Priya Gupta

Priya Gupta


Priya Gupta is a software engineer on the TensorFlow team at Google, where she works on making it easier to run TensorFlow in a distributed environment. She’s passionate about technology and education and wants machine learning to be accessible to everyone. Previously, she worked at Coursera and on the mobile ads team at Google.