-
Book Overview & Buying
-
Table Of Contents
Hands-On Artificial Intelligence for IoT - Second Edition
By :
In this chapter, we explored the implementation of distributed training using Keras 3, focusing on both data and model parallelism. We began by setting up the environment with JAX as the backend, identifying available GPUs, and configuring a DeviceMesh to manage distribution. Using a LayoutMap, we tailored how model components were sharded across devices. We then defined, compiled, and trained a neural network model using synthetic data, leveraging the power of parallel processing for efficient training across multiple GPUs.
In the next chapter, we will explore various AI cloud platforms tailored for IoT.
Join our community’s Discord space for discussions with the authors and other readers: