Skip to Content
Scaling Machine Learning with Spark
book

Scaling Machine Learning with Spark

by Adi Polak
March 2023
Intermediate to advanced
291 pages
8h 54m
English
O'Reilly Media, Inc.
Content preview from Scaling Machine Learning with Spark

Chapter 9. PyTorch Distributed Machine Learning Approach

PyTorch is an open source machine learning library developed by Facebook’s AI Research (FAIR) team and later donated to the Linux Foundation. It was designed to simplify the creation of artificial neural networks and enable applications such as computer vision, natural language processing, and more. The primary interface to PyTorch is Python, but it’s built on low-level C and C++ code. This is a very different approach from Spark, which uses Scala and Java (JVM-based programming languages) at its core.

In the previous chapters, you’ve learned about the building blocks of the machine learning workflow. We started with Spark, then expanded to explore TensorFlow’s distributed training capabilities. In this chapter, we will turn our attention to PyTorch. The goal is to help you better understand what PyTorch is and how its distributed machine learning training works, from an architectural and conceptual perspective, so you can make better decisions when combining multiple frameworks together in a distributed setting.

We will also go through a step-by-step example of working with distributed PyTorch while leveraging the previous work we did with Spark in Chapters 4 and 5 and Petastorm in Chapter 7.

This chapter covers the following:

  • A quick overview of PyTorch basics

  • PyTorch distributed strategies for training models

  • How to load Parquet data into PyTorch

  • Putting it all together—from Petastorm to a model with PyTorch

Become an O’Reilly member and get unlimited access to this title plus top books and audiobooks from O’Reilly and nearly 200 top publishers, thousands of courses curated by job role, 150+ live events each month,
and much more.

Read now

Unlock full access

More than 5,000 organizations count on O’Reilly

AirBnbBlueOriginElectronic ArtsHomeDepotNasdaqRakutenTata Consultancy Services

QuotationMarkO’Reilly covers everything we've got, with content to help us build a world-class technology community, upgrade the capabilities and competencies of our teams, and improve overall team performance as well as their engagement.
Julian F.
Head of Cybersecurity
QuotationMarkI wanted to learn C and C++, but it didn't click for me until I picked up an O'Reilly book. When I went on the O’Reilly platform, I was astonished to find all the books there, plus live events and sandboxes so you could play around with the technology.
Addison B.
Field Engineer
QuotationMarkI’ve been on the O’Reilly platform for more than eight years. I use a couple of learning platforms, but I'm on O'Reilly more than anybody else. When you're there, you start learning. I'm never disappointed.
Amir M.
Data Platform Tech Lead
QuotationMarkI'm always learning. So when I got on to O'Reilly, I was like a kid in a candy store. There are playlists. There are answers. There's on-demand training. It's worth its weight in gold, in terms of what it allows me to do.
Mark W.
Embedded Software Engineer

You might also like

Deep Learning with PyTorch

Deep Learning with PyTorch

Eli Stevens, Luca Pietro Giovanni Antiga, Thomas Viehmann
Machine Learning for High-Risk Applications

Machine Learning for High-Risk Applications

Patrick Hall, James Curtis, Parul Pandey

Publisher Resources

ISBN: 9781098106812Errata Page