Sign-In
Cart
Google JAX Cookbook: Perform machine learning and numerical computing with combined capabilities of TensorFlow and NumPy
Paperback
Publisher Price: $69.99
ISBN13: 9788197950414
Publisher: Gitforgits
Published: Oct 30 2024
Pages: 252
Weight: 0.97
Height: 0.53 Width: 7.50 Depth: 9.25
Language: English
This is the practical, solution-oriented book for every data scientists, machine learning engineers, and AI engineers to utilize the most of Google JAX for efficient and advanced machine learning. It covers essential tasks, troubleshooting scenarios, and optimization techniques to address common challenges encountered while working with JAX across machine learning and numerical computing projects.
The book starts with the move from NumPy to JAX. It introduces the best ways to speed up computations, handle data types, generate random numbers, and perform in-place operations. It then shows you how to use profiling techniques to monitor computation time and device memory, helping you to optimize training and performance. The debugging section provides clear and effective strategies for resolving common runtime issues, including shape mismatches, NaNs, and control flow errors. The book goes on to show you how to master Pytrees for data manipulation, integrate external functions through the Foreign Function Interface (FFI), and utilize advanced serialization and type promotion techniques for stable computations.
If you want to optimize training processes, this book has you covered. It includes recipes for efficient data loading, building custom neural networks, implementing mixed precision, and tracking experiments with Penzai. You'll learn how to visualize model performance and monitor metrics to assess training progress effectively. The recipes in this book tackle real-world scenarios and give users the power to fix issues and fine-tune models quickly.
Key LearningsGet your calculations done faster by moving from NumPy to JAX's optimized framework.
Make your training pipelines more efficient by profiling how long things take and how much memory they use.
Use debugging techniques to fix runtime issues like shape mismatches and numerical instability.
Get to grips with Pytrees for managing complex, nested data structures across various machine learning tasks.
Use JAX's Foreign Function Interface (FFI) to bring in external functions and give your computational capabilities a boost.
Take advantage of mixed-precision training to speed up neural network computations without sacrificing model accuracy.
Keep your experiments on track with Penzai. This lets you reproduce results and monitor key metrics.
Create your own neural networks and optimizers directly in JAX so you have full control of the architecture.
Use serialization techniques to save, load, and transfer models and training checkpoints efficiently.
Table of ContentTransition NumPy to JAX
Profiling Computation and Device Memory
Debugging Runtime Values and Errors
Mastering Pytrees for Data Structures
Exporting and Serialization
Type Promotion Semantics and Mixed Precision
Integrating Foreign Functions (FFI)
Training Neural Networks with JAX
Also in
General Computers
All Data Are Local: Thinking Critically in a Data-Driven Society
Loukissas, Yanni Alexander
Paperback
Comptia Network+ Certmike: Prepare. Practice. Pass the Test! Get Certified!: Exam N10-008
Chapple, Mike
Zacker, Craig
Paperback
Interactive Theorem Proving and Program Development: Coq'art: The Calculus of Inductive Constructions
Castéran, Pierre
Bertot, Yves
Hardcover
R for Data Science: Import, Tidy, Transform, Visualize, and Model Data
Wickham, Hadley
Grolemund, Garrett
Paperback
The Reverse Centaur's Guide to Life After AI: How to Think about Artificial Intelligence--Before It's Too Late
Doctorow, Cory
Paperback
The Agi Chronicles: The Inside Story of the Race to Create an Artificial Superintelligence
Roose, Kevin
Hardcover
Designing Data-Intensive Applications: The Big Ideas Behind Reliable, Scalable, and Maintainable Systems
Kleppmann, Martin
Riccomini, Chris
Paperback
The Year in Tech, 2027: The Insights You Need from Harvard Business Review
Chamorro-Premuzic, Tomas
Review, Harvard Business
Davenport, Thomas H.
Paperback
AI for Good: How Real People Are Using Artificial Intelligence to Fix Things That Matter
Tyrangiel, Josh
Hardcover
If Anyone Builds It, Everyone Dies: Why Superhuman AI Would Kill Us All
Soares, Nate
Yudkowsky, Eliezer
Hardcover
Human Raised: Nurturing Connection, Curiosity & Lifelong Learning in the Age of AI
Suskind, Dana, MD
Hardcover
The Eureka Machine: Why AI Is the Key to Unlocking a New Era of Scientific Discoveries
Socher, Richard
Hardcover
Hbr's 10 Must Reads on AI (with Bonus Article How to Win with Machine Learning by Ajay Agrawal, Joshua Gans, and AVI Goldfarb)
Davenport, Thomas H.
Iansiti, Marco
Review, Harvard Business
Paperback
The Year in Tech, 2026: The Insights You Need from Harvard Business Review
Review, Harvard Business
Anthony, Scott D.
Webb, Amy
Paperback
CompTIA Network+ Certification All-in-One Exam Guide, Ninth Edition (Exam N10-009)
Weissman, Jonathan S.
Paperback
The AI Con: How to Fight Big Tech's Hype and Create the Future We Want
Hanna, Alex
Bender, Emily M.
Hardcover
Fans First: Change The Game, Break the Rules & Create an Unforgettable Experience
Cole, Jesse
Paperback
Generative Ai: The Insights You Need from Harvard Business Review
Mollick, Ethan
Cremer, David De
Review, Harvard Business
Paperback
A Giant Leap: How AI Is Transforming Healthcare and What That Means for Our Future
Wachter, Robert
Hardcover
AI Literacy in Any Class: Preparing Students to Thrive in an AI Future in Any Grade Level or Content Area
Miller, Matt
Paperback
Obsolete: The AI Industry's Trillion Dollar Race to Replace Us--And How to Stop It
Lovely, Garrison
Paperback
Fundamentals of Data Engineering: Plan and Build Robust Data Systems
Reis, Joe
Housley, Matt
Paperback
Minecraft: Guide Collection 4-Book Boxed Set (Updated): Survival (Updated), Creative (Updated), Redstone (Updated), Combat
Mojang Ab
The Official Minecraft Team
Hardcover
Love Machines: How Artificial Intelligence Is Transforming Our Relationships
Muldoon, James
Hardcover
Atlas of AI: Power, Politics, and the Planetary Costs of Artificial Intelligence
Crawford, Kate
Paperback
AI Snake Oil: What Artificial Intelligence Can Do, What It Can't, and How to Tell the Difference
Kapoor, Sayash
Narayanan, Arvind
Paperback
AI with Intention: Principles and Action Steps for Teachers and School Leaders
Frontier, Tony
Paperback
Python Crash Course, 3rd Edition: A Hands-On, Project-Based Introduction to Programming
Matthes, Eric
Paperback
AP Computer Science Principles Premium, 2027: 6 Practice Tests + Comprehensive Review + Online Practice
Reichelson, Seth
Paperback
An Illustrated Guide to AI Agents: Concepts and Code for Building Agents with Llms, Tools, and Memory
Alammar, Jay
Grootendorst, Maarten
Paperback
Hbr's 10 Must Reads on Artificial Intelligence, Updated and Expanded (Featuring How AI Can Help Managers Think Through Problems by Elisa Farri and Gab
Agrawal, Ajay
Review, Harvard Business
Davenport, Thomas H.
Paperback
The Thinking Machine: Jensen Huang, Nvidia, and the World's Most Coveted Microchip
Witt, Stephen
Hardcover
CompTIA Tech+ Certification All-in-One Exam Guide, Third Edition (Exam FC0-U71)
Wempen, Faithe
Paperback
Hands-On Machine Learning with Scikit-Learn and Pytorch: Concepts, Tools, and Techniques to Build Intelligent Systems
Géron, Aurélien
Paperback
The Black Swan: Second Edition: The Impact of the Highly Improbable: With a New Section: On Robustness and Fragility
Taleb, Nassim Nicholas
Paperback
Embedded Systems with ARM Cortex-M Microcontrollers in Assembly Language and C: Fourth Edition
Zhu, Yifeng
Paperback
We Are as Gods: A Survival Guide for the Age of Abundance
Diamandis, Peter H.
Kotler, Steven
Hardcover
Artificial Intelligence: A Guide for Thinking Humans (with a New Preface)
Mitchell, Melanie
Paperback
AP Computer Science a Premium, 13th Edition: Prep Book with 6 Practice Tests + Comprehensive Review + Online Practice (2026)
Teukolsky, Roselyn, M. S.
Carman, Chris
Paperback
Hands-On Large Language Models: Language Understanding and Generation
Grootendorst, Maarten
Alammar, Jay
Paperback
Human + Machine, Updated and Expanded: Reimagining Work in the Age of AI
Wilson, H. James
Daugherty, Paul R.
Hardcover
Guilty Pleasure: The Pornification of the Internet, the Economy, and Everything Else
Nilsson, Patricia
Hardcover
Fundamentals of Software Architecture: A Modern Engineering Approach
Richards, Mark
Ford, Neal
Paperback
Power and Prediction: The Disruptive Economics of Artificial Intelligence
Gans, Joshua
Goldfarb, Avi
Agrawal, Ajay
Hardcover
The Empty Center: Permission, Connection, and What We Build When the Future is Uncertain
Drumwright, Hayes
Hardcover
The Problem with Personalization: How Advertisers Learned to Make and Break Us from Ancient Times to the AI Age
Turow, Joseph
Hardcover
Tubes: A Journey to the Center of the Internet with a New Introduction by the Author
Blum, Andrew
Paperback
Minecraft: Exploded Builds: Medieval Fortress: An Official Mojang Book
The Official Minecraft Team
Mojang Ab
Paperback
AI Projects with Raspberry Pi: High-Performance Artificial Intelligence for Robotics, Security, Home Automation, and Vision
Jepson, Brian
Hattersley, Lucy
Paperback
RHCSA Red Hat Enterprise Linux 10: Training and Exam Preparation Guide (EX200), Fourth Edition
Ghori, Asghar
Paperback
You Look Like a Thing and I Love You: How Artificial Intelligence Works and Why It's Making the World a Weirder Place
Shane, Janelle
Paperback
The Empty Center: Permission, Connection, and What We Build When the Future is Uncertain
Drumwright, Hayes
Paperback
The Coming Wave: Technology, Power, and the Twenty-First Century's Greatest Dilemma
Suleyman, Mustafa
Hardcover
Designing Machine Learning Systems: An Iterative Process for Production-Ready Applications
Huyen, Chip
Paperback
Site Reliability Engineering: How Google Runs Production Systems
Jones, Chris
Leng, Christof
Beyer, Betsy
Paperback
AI Snake Oil: What Artificial Intelligence Can Do, What It Can't, and How to Tell the Difference
Kapoor, Sayash
Narayanan, Arvind
Hardcover
Artificial Intelligence All-In-One for Dummies
Minnick, Chris
Mueller, John Paul
Massaron, Luca
Paperback
The Death of Expertise: The Campaign Against Established Knowledge and Why It Matters
Nichols, Tom
Paperback
Graphic Artists Guild Handbook, 17th Edition: Pricing & Ethical Guidelines
The Graphic Artists Guild
Paperback
The Devops Handbook, 2nd Edition: How to Create World-Class Agility, Reliability, & Security in Technology Organizations
Debois, Patrick
Kim, Gene
Humble, Jez
Paperback
The Return on AI: From Promise to Profit in the Age of Intelligent Business
Sawhney, Mohanbir
Mittal, Ashwin
Paperback
Minecraft: Roll for Adventure: The Temple of the Charged Creeper
Forbeck, Matt
Forbeck, Marty
Hardcover
Comptia A+ Complete Study Guide, 2-Volume Set: Volume 1 Core 1 Exam 220-1201 and Volume 2 Core 2 Exam 220-1202
Docter, Quentin
Buhagiar, Jon
Paperback
Navigeren door het labyrint: Een handleiding voor het beheer van data
Sebastian-Coleman, Laura
Paperback
AI for Educators: Learning Strategies, Teacher Efficiencies, and a Vision for an Artificial Intelligence Future
Miller, Matt
Paperback
R for Data Science: Import, Tidy, Transform, Visualize, and Model Data
Grolemund, Garrett
Cetinkaya-Rundel, Mine
Wickham, Hadley
Paperback
What Is Intelligence?: Lessons from AI about Evolution, Computing, and Minds
Aguera Y. Arcas, Blaise
Paperback
More Human: How the Power of AI Can Transform the Way You Lead
Carter, Jacqueline
Hougaard, Rasmus
Hardcover
Evals for AI Engineers: Systematically Measuring and Improving AI Applications
Husain, Hamel
Shankar, Shreya
Paperback