{"product_id":"deep-learning-with-jax-9781633438880","title":"Deep Learning with Jax","description":"\u003cb\u003eAccelerate deep learning and other number-intensive tasks with JAX, Google's awesome high-performance numerical computing library.\u003c\/b\u003e \u003cp\u003e\u003c\/p\u003eThe 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. \u003cp\u003e\u003c\/p\u003e In \u003ci\u003eDeep Learning with JAX\u003c\/i\u003e you will learn how to: \u003cp\u003e\u003c\/p\u003e - Use JAX for numerical calculations\u003cbr\u003e - Build differentiable models with JAX primitives\u003cbr\u003e - Run distributed and parallelized computations with JAX\u003cbr\u003e - Use high-level neural network libraries such as Flax\u003cbr\u003e - Leverage libraries and modules from the JAX ecosystem \u003cp\u003e\u003c\/p\u003e \u003ci\u003eDeep Learning with JAX\u003c\/i\u003e 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. \u003cp\u003e\u003c\/p\u003e Purchase of the print book includes a free eBook in PDF and ePub formats from Manning Publications. \u003cp\u003e\u003c\/p\u003e \u003cb\u003eAbout the technology\u003c\/b\u003e \u003cp\u003e\u003c\/p\u003e 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. \u003cp\u003e\u003c\/p\u003e \u003cb\u003eAbout the book\u003c\/b\u003e \u003cp\u003e\u003c\/p\u003e \u003ci\u003eDeep Learning with JAX\u003c\/i\u003e 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. \u003cp\u003e\u003c\/p\u003e \u003cb\u003eWhat's inside\u003c\/b\u003e \u003cp\u003e\u003c\/p\u003e - Use JAX for numerical calculations\u003cbr\u003e - Build differentiable models with JAX primitives\u003cbr\u003e - Run distributed and parallelized computations with JAX\u003cbr\u003e - Use high-level neural network libraries such as Flax \u003cp\u003e\u003c\/p\u003e \u003cb\u003eAbout the reader\u003c\/b\u003e \u003cp\u003e\u003c\/p\u003e For intermediate Python programmers who are familiar with deep learning. \u003cp\u003e\u003c\/p\u003e \u003cb\u003eAbout the author\u003c\/b\u003e \u003cp\u003e\u003c\/p\u003e \u003cb\u003eGrigory Sapunov\u003c\/b\u003e holds a Ph.D. in artificial intelligence and is a Google Developer Expert in Machine Learning. \u003cp\u003e\u003c\/p\u003e The technical editor on this book was \u003cb\u003eNicholas McGreivy\u003c\/b\u003e. \u003cp\u003e\u003c\/p\u003e \u003cb\u003eTable of Contents\u003c\/b\u003e\u003cbr\u003e Part 1\u003cbr\u003e 1 When and why to use JAX\u003cbr\u003e 2 Your first program in JAX\u003cbr\u003e Part 2\u003cbr\u003e 3 Working with arrays\u003cbr\u003e 4 Calculating gradients\u003cbr\u003e 5 Compiling your code\u003cbr\u003e 6 Vectorizing your code\u003cbr\u003e 7 Parallelizing your computations\u003cbr\u003e 8 Using tensor sharding\u003cbr\u003e 9 Random numbers in JAX\u003cbr\u003e 10 Working with pytrees\u003cbr\u003e Part 3\u003cbr\u003e 11 Higher-level neural network libraries\u003cbr\u003e 12 Other members of the JAX ecosystem\u003cbr\u003e A Installing JAX\u003cbr\u003e B Using Google Colab\u003cbr\u003e C Using Google Cloud TPUs\u003cbr\u003e D Experimental parallelization\u003cbr\u003e\u003cbr\u003e\u003cbr\u003e\u003cb\u003eAbout the Author\u003c\/b\u003e\u003cbr\u003e\u003cb\u003eGrigory Sapunov\u003c\/b\u003e is a co-founder and CTO of Intento. He is a software engineer with more than twenty years of experience. Grigory holds a Ph.D. in artificial intelligence and is a Google Developer Expert in Machine Learning.\u003cbr\u003e","brand":"Manning Publications","offers":[{"title":"Default Title","offer_id":50901965046034,"sku":"9781633438880","price":55.99,"currency_code":"USD","in_stock":false}],"thumbnail_url":"\/\/cdn.shopify.com\/s\/files\/1\/0831\/4771\/8930\/files\/img_ae8b4e36-57b6-4ca3-9448-d0b32fbdf6ec.jpg?v=1738443602","url":"https:\/\/surprise-castle.myshopify.com\/products\/deep-learning-with-jax-9781633438880","provider":"Surprise Castle","version":"1.0","type":"link"}