Oreilly – Deep Learning with JAX, Video Edition 2024-11

Description

Deep Learning with JAX, Video Edition. This course teaches participants how to accelerate high-speed numerical computation and deep learning using Google’s JAX library. The JAX library addresses fundamental performance challenges at the heart of deep learning and scientific computing by combining Google’s accelerated linear algebra platform with an optimized version of NumPy, delivering dramatic increases in performance and low-level transformations. As a hands-on guide, this course builds on the understanding of JAX concepts through concrete examples by Google expert Grigory Saponov. JAX makes model training and inference fast and efficient, especially on large datasets, by providing fine-grained control over processes such as gradient computation. With a robust ecosystem, the library enables evolutionary computing, federated learning, and other performance-sensitive tasks. During this course, participants will learn how unique JAX features such as functional programming help solve performance challenges such as distributing computations across a cluster of tensor processing units by building an image classifier, an image filter application, and other practical projects. They will also learn how to leverage the JAX ecosystem, including high-level libraries, as well as its integration with TensorFlow and PyTorch for data loading and deployment.

What you will learn

• Using JAX for numerical calculations.
• Building Differentiable Models with JAX Primitives.
• Implementing distributed and parallelized computations with JAX.
• Using high-level neural network libraries such as Flax.
• Optimized use of libraries and modules from the JAX ecosystem.
• How to build effective neural networks with JAX.

This course is suitable for people who:

• Intermediate Python programmers who are familiar with deep learning.

Course details for Deep Learning with JAX, Video Edition

• Publisher:  Oreilly
• Instructor: Grigory Sapunov
• Training level: Beginner to advanced
• Training duration: 10 hours and 58 minutes

Course headings

Course images