Get Discount 5% Off
Subscribe to our newsletters now and stay up-to-date with new arrivals, updates and deals.
Deep Learning with Jax
Accelerate deep learning and other number-intensive tasks with JAX, Google's awesome high-performance numerical computing library. The JAX...
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
SKU:
9781633438880
Product Type:
Books
Please hurry! Only 72 left in stock
10 customers are viewing this product
Couldn't load pickup availability
Accelerate deep learning and other number-intensive tasks with JAX, Google's awesome high-performance numerical computing library. The JAX numerical computing library tackles the core performance challenges at the heart of deep learning and other scientific computing tasks. By combining Google's Accelerated Linear Algebra platform (XLA) with a hyper-optimized version of NumPy and a variety of other high-performance features, JAX delivers a huge performance boost in low-level computations and transformations. In Deep Learning with JAX you will learn how to: - Use JAX for numerical calculations
- Build differentiable models with JAX primitives
- Run distributed and parallelized computations with JAX
- Use high-level neural network libraries such as Flax
- Leverage libraries and modules from the JAX ecosystem Deep Learning with JAX is a hands-on guide to using JAX for deep learning and other mathematically-intensive applications. Google Developer Expert Grigory Sapunov steadily builds your understanding of JAX's concepts. The engaging examples introduce the fundamental concepts on which JAX relies and then show you how to apply them to real-world tasks. You'll learn how to use JAX's ecosystem of high-level libraries and modules, and also how to combine TensorFlow and PyTorch with JAX for data loading and deployment. Purchase of the print book includes a free eBook in PDF and ePub formats from Manning Publications. About the technology Google's JAX offers a fresh vision for deep learning. This powerful library gives you fine control over low level processes like gradient calculations, delivering fast and efficient model training and inference, especially on large datasets. JAX has transformed how research scientists approach deep learning. Now boasting a robust ecosystem of tools and libraries, JAX makes evolutionary computations, federated learning, and other performance-sensitive tasks approachable for all types of applications. About the book Deep Learning with JAX teaches you to build effective neural networks with JAX. In this example-rich book, you'll discover how JAX's unique features help you tackle important deep learning performance challenges, like distributing computations across a cluster of TPUs. You'll put the library into action as you create an image classification tool, an image filter application, and other realistic projects. The nicely-annotated code listings demonstrate how JAX's functional programming mindset improves composability and parallelization. What's inside - Use JAX for numerical calculations
- Build differentiable models with JAX primitives
- Run distributed and parallelized computations with JAX
- Use high-level neural network libraries such as Flax About the reader For intermediate Python programmers who are familiar with deep learning. About the author Grigory Sapunov holds a Ph.D. in artificial intelligence and is a Google Developer Expert in Machine Learning. The technical editor on this book was Nicholas McGreivy. Table of Contents
Part 1
1 When and why to use JAX
2 Your first program in JAX
Part 2
3 Working with arrays
4 Calculating gradients
5 Compiling your code
6 Vectorizing your code
7 Parallelizing your computations
8 Using tensor sharding
9 Random numbers in JAX
10 Working with pytrees
Part 3
11 Higher-level neural network libraries
12 Other members of the JAX ecosystem
A Installing JAX
B Using Google Colab
C Using Google Cloud TPUs
D Experimental parallelization
Author: Grigory Sapunov
Publisher: Manning Publications
Published: 10/29/2024
Pages: 408
Binding Type: Paperback
Weight: 1.63lbs
Size: 9.20h x 7.30w x 0.90d
ISBN: 9781633438880
- Build differentiable models with JAX primitives
- Run distributed and parallelized computations with JAX
- Use high-level neural network libraries such as Flax
- Leverage libraries and modules from the JAX ecosystem Deep Learning with JAX is a hands-on guide to using JAX for deep learning and other mathematically-intensive applications. Google Developer Expert Grigory Sapunov steadily builds your understanding of JAX's concepts. The engaging examples introduce the fundamental concepts on which JAX relies and then show you how to apply them to real-world tasks. You'll learn how to use JAX's ecosystem of high-level libraries and modules, and also how to combine TensorFlow and PyTorch with JAX for data loading and deployment. Purchase of the print book includes a free eBook in PDF and ePub formats from Manning Publications. About the technology Google's JAX offers a fresh vision for deep learning. This powerful library gives you fine control over low level processes like gradient calculations, delivering fast and efficient model training and inference, especially on large datasets. JAX has transformed how research scientists approach deep learning. Now boasting a robust ecosystem of tools and libraries, JAX makes evolutionary computations, federated learning, and other performance-sensitive tasks approachable for all types of applications. About the book Deep Learning with JAX teaches you to build effective neural networks with JAX. In this example-rich book, you'll discover how JAX's unique features help you tackle important deep learning performance challenges, like distributing computations across a cluster of TPUs. You'll put the library into action as you create an image classification tool, an image filter application, and other realistic projects. The nicely-annotated code listings demonstrate how JAX's functional programming mindset improves composability and parallelization. What's inside - Use JAX for numerical calculations
- Build differentiable models with JAX primitives
- Run distributed and parallelized computations with JAX
- Use high-level neural network libraries such as Flax About the reader For intermediate Python programmers who are familiar with deep learning. About the author Grigory Sapunov holds a Ph.D. in artificial intelligence and is a Google Developer Expert in Machine Learning. The technical editor on this book was Nicholas McGreivy. Table of Contents
Part 1
1 When and why to use JAX
2 Your first program in JAX
Part 2
3 Working with arrays
4 Calculating gradients
5 Compiling your code
6 Vectorizing your code
7 Parallelizing your computations
8 Using tensor sharding
9 Random numbers in JAX
10 Working with pytrees
Part 3
11 Higher-level neural network libraries
12 Other members of the JAX ecosystem
A Installing JAX
B Using Google Colab
C Using Google Cloud TPUs
D Experimental parallelization
Author: Grigory Sapunov
Publisher: Manning Publications
Published: 10/29/2024
Pages: 408
Binding Type: Paperback
Weight: 1.63lbs
Size: 9.20h x 7.30w x 0.90d
ISBN: 9781633438880
Returns Policy
You may return most new, unopened items within 30 days of delivery for a full refund. We'll also pay the return shipping costs if the return is a result of our error (you received an incorrect or defective item, etc.).
You should expect to receive your refund within four weeks of giving your package to the return shipper, however, in many cases you will receive a refund more quickly. This time period includes the transit time for us to receive your return from the shipper (5 to 10 business days), the time it takes us to process your return once we receive it (3 to 5 business days), and the time it takes your bank to process our refund request (5 to 10 business days).
If you need to return an item, simply login to your account, view the order using the "Complete Orders" link under the My Account menu and click the Return Item(s) button. We'll notify you via e-mail of your refund once we've received and processed the returned item.
Shipping
We can ship to virtually any address in the world. Note that there are restrictions on some products, and some products cannot be shipped to international destinations.
When you place an order, we will estimate shipping and delivery dates for you based on the availability of your items and the shipping options you choose. Depending on the shipping provider you choose, shipping date estimates may appear on the shipping quotes page.
Please also note that the shipping rates for many items we sell are weight-based. The weight of any such item can be found on its detail page. To reflect the policies of the shipping companies we use, all weights will be rounded up to the next full pound.
Related Products
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Recently Viewed Products
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
Example product title
- $181.15 AUD
- $181.15 AUD
- Unit price
- per
- Choosing a selection results in a full page refresh.