"Google JAX Essentials" is a comprehensive guide designed for machine learning and deep learning professionals aiming to leverage the power and capabilities of Google's JAX library in their projects. Over the course of eight chapters, this book takes the reader from understanding the challenges of deep learning and numerical computations in the existing frameworks to the essentials of Google JAX, its functionalities, and how to leverage it in real-world machine learning and deep learning projects. The book starts by ...
Read More
"Google JAX Essentials" is a comprehensive guide designed for machine learning and deep learning professionals aiming to leverage the power and capabilities of Google's JAX library in their projects. Over the course of eight chapters, this book takes the reader from understanding the challenges of deep learning and numerical computations in the existing frameworks to the essentials of Google JAX, its functionalities, and how to leverage it in real-world machine learning and deep learning projects. The book starts by emphasizing the importance of numerical computing in ML and DL, demonstrating the limitations of standard libraries like NumPy, and introducing the solution offered by JAX. It then guides the reader through the installation of JAX on different computing environments like CPUs, GPUs, and TPUs, and its integration into existing ML and DL projects. The book details the advanced numerical operations and unique features of JAX, including JIT compilation, automatic differentiation, batched operations, and custom gradients. It illustrates how these features can be employed to write code that is both simpler and faster. The book also delves into parallel computation, the effective use of the vmap function, and the use of pmap for distributed computing. Lastly, the reader is walked through the practical application of JAX in training different deep learning models, including RNNs, CNNs, and Bayesian models, with an additional focus on performance-tuning strategies for JAX applications. Key Learnings Mastering the installation and configuration of JAX on various computing environments. Understanding the intricacies of JAX's advanced numerical operations. Harnessing the power of JIT compilation in JAX for accelerated computations. Implementing batched operations using the vmap function for efficient processing. Leveraging automatic differentiation and custom gradients in JAX. Proficiency in using the pmap function for distributed computing in JAX. Training different types of deep learning models using JAX. Applying performance tuning strategies to maximize JAX application efficiency. Integrating JAX into existing machine learning and deep learning projects. Complementing the official JAX documentation with practical, real-world applications. Table of Content Necessity for Google JAX Unravelling JAX Setting up JAX for Machine Learning and Deep Learning JAX for Numerical Computing Diving Deeper into Auto Differentiation and Gradients Efficient Batch Processing with JAX Power of Parallel Computing with JAX Training Neural Networks with JAX Audience This is must read for machine learning and deep learning professionals to be skilled with the most innovative deep learning library. Knowing Python and experience with machine learning is sufficient is desired to begin with this book
Read Less
Choose your shipping method in Checkout. Costs may vary based on destination.
Seller's Description:
New. Trade paperback (US). Glued binding. 120 p. In Stock. 100% Money Back Guarantee. Brand New, Perfect Condition, allow 4-14 business days for standard shipping. To Alaska, Hawaii, U.S. protectorate, P.O. box, and APO/FPO addresses allow 4-28 business days for Standard shipping. No expedited shipping. All orders placed with expedited shipping will be cancelled. Over 3, 000, 000 happy customers.
Choose your shipping method in Checkout. Costs may vary based on destination.
Seller's Description:
PLEASE NOTE, WE DO NOT SHIP TO DENMARK. New Book. Shipped from UK in 4 to 14 days. Established seller since 2000. Please note we cannot offer an expedited shipping service from the UK.
Choose your shipping method in Checkout. Costs may vary based on destination.
Seller's Description:
Fine. Trade paperback (US). Glued binding. 120 p. In Stock. 100% Money Back Guarantee. Brand New, Perfect Condition, allow 4-14 business days for standard shipping. To Alaska, Hawaii, U.S. protectorate, P.O. box, and APO/FPO addresses allow 4-28 business days for Standard shipping. No expedited shipping. All orders placed with expedited shipping will be cancelled. Over 3, 000, 000 happy customers.
Choose your shipping method in Checkout. Costs may vary based on destination.
Seller's Description:
PLEASE NOTE, WE DO NOT SHIP TO DENMARK. New Book. Shipped from UK in 4 to 14 days. Established seller since 2000. Please note we cannot offer an expedited shipping service from the UK.
Choose your shipping method in Checkout. Costs may vary based on destination.
Seller's Description:
New. Trade paperback (US). Glued binding. 120 p. In Stock. 100% Money Back Guarantee. Brand New, Perfect Condition, allow 4-14 business days for standard shipping. To Alaska, Hawaii, U.S. protectorate, P.O. box, and APO/FPO addresses allow 4-28 business days for Standard shipping. No expedited shipping. All orders placed with expedited shipping will be cancelled. Over 3, 000, 000 happy customers.
Choose your shipping method in Checkout. Costs may vary based on destination.
Seller's Description:
Fine. Trade paperback (US). Glued binding. 120 p. In Stock. 100% Money Back Guarantee. Brand New, Perfect Condition, allow 4-14 business days for standard shipping. To Alaska, Hawaii, U.S. protectorate, P.O. box, and APO/FPO addresses allow 4-28 business days for Standard shipping. No expedited shipping. All orders placed with expedited shipping will be cancelled. Over 3, 000, 000 happy customers.
