
Автор: Mei Wong
Издательство: GitforGits
Год: 2023
Страниц: 225
Язык: английский
Формат: pdf, epub, mobi
Размер: 10.1 MB
"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.